weixin_43897346的博客

运行程序显示不存在torchvision module? 这是因为还没有安装pytorch框架。 首先查看你的cuda版本。 nvidia-smi//查看gpu 这里可以看到cuda是版本为10.1. 登录pytorch官网https://pytorch.org/ 选择pytorch版本，Linux,Conda命令安装，Python版， CUDA10.1,就能看到对应的安装命令啦。 输入对应安装命令，选择y进行安装。 conda install pytorch torchvision cudatoo