pytorch_gpu 安装

pytorch_gpu 安装

参考博客:

https://blog.csdn.net/weixin_40431584/article/details/105119633
博客主要是在pip install 中信任阿里源速度起飞,还是很棒

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

当出现安装过程中出现,连接断开的操作时,处理方案。记得加入 --default-timeout=1000

pip3 --default-timeout=1000 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

pytorch 官网进不去解决

1、使用下载链接,进去查看自己的版本:https://download.pytorch.org/whl/torch_stable.html 这个网站还可以进去
2、使用清华镜像进行查看
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
清华镜像更加全面,并且支持低版本的torch+高版本的CUDA 配置

torch 版本切换

注意这里使用pip 是不行的 使用conda 就更好,有些pip 中cuda不兼容的情况,比如1.2 torch后面就开始不兼容 cuda10.0 这个很恼火,解决方案,建议使用 清华的更改cuda【简而言之就是用pip 直接装torch那一套,自动安装新的cuda】,实在不行,就用1.2.0也可以,需要在清华镜像下载。

(pytorch_gpu) C:\Users\MaoHu>conda install --use-local E:\anconda\pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2

验证torch 版本和是否支持gpu

>>> import torch
>>> print(torch.__version__,torch.cuda.is_available())
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值