安装环境准备:
1.首先你需要安装Anaconda:
需要注意Anaconda自带的python版本。我安装的Anaconda自带的是python3.5版本。
2.其次你需要安装CUDA 10.0和对应的cuDNN v7(见相应的博客教程)。
安装pytorch
官网(https://pytorch.org/get-started/locally/)给出的安装命令是:
CUDA 10.0
# Python 3.5
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
上述两个命令对应的包都需要安装。实测发现由于安装包很大(698M) ,直接用如上命令安装下载时频频出错,无法安装。
正确安装方法如下:
1.离线安装上述whl文件。你可以直接将intall后面的网址输入浏览器下载,但实测下载成功率也不高。所以我将此文件上传至百度网盘:
链接:https://pan.baidu.com/s/1i3E8Na08UKSajRMKLCBrMg
提取码:amk4
下载后使用命令
pip install (whl文件的存放路径)/torch-1.0.1-cp35-cp35m-win_amd64.whl
即可安装成功。
2.实测第二条安装命令(pip3 install torchvision)也无法安装成功,显示找不到安装包。解决方法:
先更新pip:
python -m pip install --upgrade pip
随后使用清华源安装:
pip3 install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
至此pytorch就已经安装成功啦!在命令行输入如下命令测试一下:
python
from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
此命令生成一个满足0-1均匀分布的矩阵:
tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])
测试一下CUDA是否可用:
import torch #之前import过了此句可不写
torch.cuda.is_available()
显示true就表示没有问题。