Ubuntu 安装gpu版本的torch时遇到的图片问题
原因:cuda版本与pytorch版本不匹配
解决办法
1.查看torch的版本
python
import torch
torch.__version__
2.查看torch对应的cuda的版本
torch.version.cuda
2.查看cuda是否可用
torch.cuda.is_available()
3.查看cuda的版本
cat /usr/local/cuda/version.txt
4.cuda版本与torch的cuda版本不一致的话
卸载安装的torch: pip uninstall torch
找对应版本的torch 网址: https://pytorch.org/get-started/previous-versions/
安装命令也在网址里,重新安装torch
本文中的命令是:conda install pytorch=0.4.1 cuda90 -c pytorch