安装GPU版本pytorch教程(避坑)

Pytorch官方网站:

https://pytorch.org/

复制上述语句,执行,但速度超级超级慢。即便有梯子,速度依旧很慢。容易超时失败!

遇到好多好多坑之后,决定采用阿里云的源文件,即在上述代码后附上以下代码:

-i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

相当于,用阿里云的pip源并且信任这个源,可以用。

 

具体安装如下:(最好管理员身份打开)

1.更新Pip

python -m pip install --upgrade pip -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

2.安装torch 1.1.0

pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl  -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

(注:本案例python版本为3.7,其他版本如python3.6,则修改成cp36-cp36m)

3.安装torchvision 0.3.0

pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl  -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

(注意事项同上)

 

测试torch能否用GPU

>>>import torch

>>>torch.__version__

>>>torch.cuda.is_available()

输出True即可。

图上,表示安装成功啦!!!

 

 

  • 42
    点赞
  • 222
    收藏
  • 打赏
    打赏
  • 25
    评论

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

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:终极编程指南 设计师:CSDN官方博客 返回首页
评论 25

打赏作者

AI阿聪

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值