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开始页

 通过自己设备性能一一选择,最后等到相应的代码。将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

  • 10
    点赞
  • 37
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值