windows10(64位)Anaconda3+Python3.6搭建Pytorch(cpu版本)
第一步:修改默认数据源
(默认的官方数据源下载速度慢!!!修改默认数据源能让安装速度变的很快。)
注:在cmd内执行如下命令:
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
python -m pip install --upgrade pip (最好执行一下该命令,防止后面操作报错)
第二步: 新建虚拟环境
在Anaconda Prompt (Anaconda3)内执行如下命令
conda create -n pytorch pip python=3.6
注:我的python是3.6.13
第三步:激活新建的虚拟环境
activate pytorch
第四步:安装Pytorch
使用如下语句可以安装cu91版本的pytorch
1.
(1)先下载torch-0.4.0-cp36-cp36m-win_amd64.whl文件
直接点击下载.whl文件
或:
http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
(2)将下载的文件放入(随便什么盘)D盘
pip install d:\torch-0.4.0-cp36-cp36m-win_amd64.whl
2.或者直接一步到位(太慢)
pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl
注:我的CUDA版本是9.1.126,我安装的CUDA9.1与Python3.6.13
以上步骤安装完torchvision的版本是0.4.1,如果不能使用,安装torchvision0.1.8
pip install torchvision==0.1.8
第五步:测试
python(开始写python代码)
import torch
import torchvision
print(torch.__version__)
第六步:退出python
exit()
成功打印,测试成功
第七步:关闭Pytorch Anaconda虚拟环境
conda deactivate
完美安装成功!!!