anaconda安装pytorch GPU版小白教程
step1:确认自己的cuda的版本号nvcc --version例如我的,显示结果如下:由结果可以看出cuda版本号是9.0step2:去官网查看命令https://pytorch.org/注意:如果cuda 不匹配,命令是无法使用的,比如我的cuda是9.0,则命令是:conda install pytorch torchvision cudatoolkit=9.0 -c pytorchstep3:测试是否安装成功:(pytorch-3) XXXX>pythonPyth
原创
2020-10-28 16:32:33 ·
1477 阅读 ·
0 评论