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())