ubuntu16.04下pytorch安装(pip安装包失败解决办法)

配置

环境

Python	CUDA
3.6.9	10.1

安装的pytorch版本

torch	torchvision
1.2.0	0.4.0

遇到的问题

问题一：用conda安装，数据不能放上GPU

pytorch官网

选择自己需要的版本，复制命令，在终端输入

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

问题：安装好之后训练，发现数据不能传到GPU上
查找原因：conda list 中只有pytorch，没有 torch 和 torchvision

问题二：CUDA报错

卸载重新安装pytorch后运行代码
报错

RuntimeError: CUDA error: invalid device ordinal

查找原因：太复杂了，没仔细看

最终解决：用pip安装

尝试一

pip install torch