xyh_qianxiao的博客

step1:确认自己的cuda的版本号 nvcc --version 例如我的，显示结果如下： 由结果可以看出cuda版本号是9.0 step2:去官网查看命令 https://pytorch.org/ 注意：如果cuda 不匹配，命令是无法使用的，比如我的cuda是9.0，则命令是： conda install pytorch torchvision cudatoolkit=9.0 -c pytorch step3：测试是否安装成功： (pytorch-3) XXXX>python Pyth