说明:我安装的是cuda8.0+cndnn7.1+pytorch1.0.1,亲测可为,不用按照官网来,官网速度太慢了,而且会断。conda会比pip好,可自动选择适合电脑环境的版本。
0.换源,换源,换源
在Anaconda中使用清华镜像源
添加清华镜像至Anaconda仓库
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
pytorch
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
1.安装pytorch-gpu
在cmd下运行
conda install pytorch torchvision
会自动安装适合你电脑的pytorch-gpu版本