Anaconda下安装Pytorch-GPU版

设备环境:

  1. windows10  
  2. Anaconda5.2
  3. 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


Pytorch开始页

Anaconda下安装Pytorch-GPU版_第1张图片

 通过自己设备性能一一选择,最后等到相应的代码。将command复制在conda prompt 中运行。

4. 本地安装

    由于安装过程中网络的原因,在下载pytorch-0.4.1时,时间会很长,有时能会出现连接错误。所以将torch.whl文件下载到本地,分享给大家。

传送门:【百度网盘】链接:https://pan.baidu.com/s/1pSofoGl8TN5iWM-VcZFWcw 
提取码:n3z7 
 

Anaconda下安装Pytorch-GPU版_第2张图片

【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

你可能感兴趣的:(Anaconda下安装Pytorch-GPU版)