草庐IT

pytorch-cuda

全部标签

【pytorch】torch.cdist使用说明

使用说明torch.cdist的使用介绍如官网所示,它是批量计算两个向量集合的距离。其中,x1和x2是输入的两个向量集合。p默认为2,为欧几里德距离。它的功能上等同于scipy.spatial.distance.cdist(input,’minkowski’,p=p)如果x1的shape是[B,P,M],x2的shape是[B,R,M],则cdist的结果shape是[B,P,R]进一步的解释x1一般是输入矢量,而x2一般是码本。x2中所有的元素分别与x1中的每一个元素求欧几里德距离(当p默认为2时)如下面示例importtorchx1=torch.FloatTensor([0.1,0.2,0

【pytorch】torch.cdist使用说明

使用说明torch.cdist的使用介绍如官网所示,它是批量计算两个向量集合的距离。其中,x1和x2是输入的两个向量集合。p默认为2,为欧几里德距离。它的功能上等同于scipy.spatial.distance.cdist(input,’minkowski’,p=p)如果x1的shape是[B,P,M],x2的shape是[B,R,M],则cdist的结果shape是[B,P,R]进一步的解释x1一般是输入矢量,而x2一般是码本。x2中所有的元素分别与x1中的每一个元素求欧几里德距离(当p默认为2时)如下面示例importtorchx1=torch.FloatTensor([0.1,0.2,0

PyTorch model 返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()

PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()文章目录PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()前言一、model.modules(),model.children(),model.parameters()1.1model.modules()1.2model.named_modules()1.3model.children()

PyTorch model 返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()

PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()文章目录PyTorchmodel返回函数总结——model.state_dict(),model.modules(),model.children(),model.parameters()前言一、model.modules(),model.children(),model.parameters()1.1model.modules()1.2model.named_modules()1.3model.children()

【CUDA开发环境搭建】

文章目录一、VisualStudio2022+CUDA11.6(Windows10)1、VisualStudiocommunity2022的安装2、NVIDIA驱动的安装3、CUDA11.6的安装4、VisualStudio配置与测试5、debug二、VisualStudioCode+CUDA11.6(远程连接Ubuntu)1、VScode的安装。(省略)2、VScodessh远程连接(可参考其他详细教程,这里只记录主要步骤)3、安装相关插件这里是安装CUDA11.64、编译和调试(以matrixMul为例)引用记录两种CUDA开发环境的配置过程。一、VisualStudio2022+CUDA

【CUDA开发环境搭建】

文章目录一、VisualStudio2022+CUDA11.6(Windows10)1、VisualStudiocommunity2022的安装2、NVIDIA驱动的安装3、CUDA11.6的安装4、VisualStudio配置与测试5、debug二、VisualStudioCode+CUDA11.6(远程连接Ubuntu)1、VScode的安装。(省略)2、VScodessh远程连接(可参考其他详细教程,这里只记录主要步骤)3、安装相关插件这里是安装CUDA11.64、编译和调试(以matrixMul为例)引用记录两种CUDA开发环境的配置过程。一、VisualStudio2022+CUDA

Docker配置深度学习pytorch gpu环境

一、docker的概念原理参考资料:https://zhuanlan.zhihu.com/p/81693661          https://www.zhihu.com/question/506867139          https://blog.csdn.net/weixin_44751294/article/details/123191429    看过上面两个链接,就对docker是什么作用,怎么用有个大概的了解。简单摘要一些对我来说比较重要的点如下:(1)docker可以让环境配置变得简单,在多人同时使用一台服务器或者时不时换电脑换机器所有的包时都很有用(2)docker可以看

Docker配置深度学习pytorch gpu环境

一、docker的概念原理参考资料:https://zhuanlan.zhihu.com/p/81693661          https://www.zhihu.com/question/506867139          https://blog.csdn.net/weixin_44751294/article/details/123191429    看过上面两个链接,就对docker是什么作用,怎么用有个大概的了解。简单摘要一些对我来说比较重要的点如下:(1)docker可以让环境配置变得简单,在多人同时使用一台服务器或者时不时换电脑换机器所有的包时都很有用(2)docker可以看

pytorch :OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading 【已解决】

OSError:[WinError1455]页面文件太小,无法完成操作。Errorloading"D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\torch\lib\cudnn_cnn_infer64_8.dll"oroneofitsdependencies.win11系统,运行出现上述错误 解决方法 1.降低Batch_Size大小:过大的Batch_Size(人工智能术语:指一次迭代过程送入的数据集样本量)显然需要更多的内存支持,出现“页面文件太小”,可尝试降低Batch_Size大小。2.调整虚拟内存可用硬盘空间大小:虚

pytorch :OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading 【已解决】

OSError:[WinError1455]页面文件太小,无法完成操作。Errorloading"D:\ProgramData\Anaconda3\envs\openmmlab\lib\site-packages\torch\lib\cudnn_cnn_infer64_8.dll"oroneofitsdependencies.win11系统,运行出现上述错误 解决方法 1.降低Batch_Size大小:过大的Batch_Size(人工智能术语:指一次迭代过程送入的数据集样本量)显然需要更多的内存支持,出现“页面文件太小”,可尝试降低Batch_Size大小。2.调整虚拟内存可用硬盘空间大小:虚