设备环境:
- windows10
- Anaconda5.2
- python 3.6
更换Anaconda国内源-参考: Anaconda安装杂记
1. 新建Pytorch虚拟环境
conda create -n Pytorch python=3.6
# 环境名称:Pytorch python版本:3.6
2. 激活Pytorch环境
conda activate Pytorch
3. 在Pytorch环境下安装torch
通过自己设备性能一一选择,最后等到相应的代码。将command复制在conda prompt 中运行。
4. 本地安装
由于安装过程中网络的原因,在下载pytorch-0.4.1时,时间会很长,有时能会出现连接错误。所以将torch.whl文件下载到本地,分享给大家。
传送门:【百度网盘】链接:https://pan.baidu.com/s/1pSofoGl8TN5iWM-VcZFWcw
提取码:n3z7
【tips】:本地安装whl文件,首先进入文件所在的当前目录下通过运行
pip install torch-0.4.1-cp36-cp36m-win_amd64.whl
即可完成安装。
5. 安装torchvision
pip install torchvision
6. 测试
import torch # 导入torch包 如果没有报错则表示安装成功
torch.cuda.is_available() # 返回True则表示GPU可用, 反之则不可用
记于2018.12.01