Linux/Windows GPU/CPU版 pytorch安装

Posted 2020-11-20 sbj123456789

Windows上安装：

最新0.4.0版本：

在Pytorch官网 https://pytorch.org/ 上选择对应版本安装即可，conda安装的比较慢，建议选择pip安装的（虽然还是很慢），当然能找到不错的镜像也是极好的。安装CPU版的在CUDA处选None。

0.3.0等以前的老版本：

建议参考知乎https://www.zhihu.com/question/67209417，用里面的镜像会快一些。

Linux上安装：

先装好CUDA和Cudnn，参考 https://www.cnblogs.com/sbj123456789/p/9219631.html

创建环境：

　　conda create -n pytorch python=3.6

激活环境：

　　conda activate pytorch 或 source activate pytorch

退出环境：

　　conda deactivate

查看所有环境：

　　conda env list

直接pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

（直接pip install torch torchvision慢出天际）

把后面直接换成想要的版本即可，具体可参考官网。

GPU安装后的测试：

换成清华源：

直接从国外的源下载速度还是不够快，可以换成清华源。

参考：https://blog.csdn.net/zzq060143/article/details/88042075

pytorch和CUDA版本号改一下就ok

一些坑：

GPU 1080 以下的版本尽量安装0.3.0的，安装0.4.0的在Windows上会不支持。