网站怎么做推广中国最大的软件外包公司

情况1,真的没有安装pytorch

虚拟环境里面真的是没有torch,这种情况就easy job了,点击此链接直接安装与CUDA对应的pytorch版本,CTRL+F直接搜索对应CUDA版本即可查找到对应的命令.按图索骥按部就班安装即可!

情况2(安装了与CUDA不对应的pytorch版本导致无法识别出torch)

安装了与CUDA不对应的pytorch版本导致无法识别出torch
我这里首先通过

pip list | grep torch

指令搜索出pip安装包仲含有torch字样的安装包,我这里显然通过pip安装了pytoch,都是cuda 11.1版本的

但是运行py程序依然显示没有找到模块torch,我十分不解

接着我查找到了服务器的cuda版本
使用

cat ~/.bashrc
#或者使用
#nvcc -V
#也是可以的!

指令查看安装的CUDA版本,在末尾几行查找到cuda安装的版本为11.3

到这里发现安装的cuda 11.1的pytorch版本与本机安装的11.3版本cuda不对应,因此重新安装了虚拟环境的pytorch!
最后成功导入torch不报错!