pytorch，torch，torchvision的gpu版本安装避坑

在线or离线安装

最近使用pytorch跑深度学习模型，需要用到与gpu相对应的包，写此博文，提醒后来者避避坑

pytorch

本次使用python3.6（conda环境），一般conda环境可以创建多个Python版本，互不干扰，比单一Python要好用，在pycharm上添加pytorch库。

点击左下方 + ，进去页面输入pytorch，安装对应的版本

torch和torchvision

torch，torchvision库在pycharm中没搜到，在conda环境下激活python36，使用pip命令安装
在使用计算机的狗之前，要安装好显卡驱动，cuda和处cudnn，请至
https://blog.csdn.net/cashmood/article/details/105081586

因为要是用gpu跑代码，所以和torch，torchvision相关的包都必须是和gpu相对应的，

（1）离线安装（强烈推荐）

下载对应的torch和torchvision ：http://download.pytorch.org/whl/torch_stable.html

或者：（第三方扩展包）https://www.lfd.uci.edu/~gohlke/pythonlibs/#pyhook
或者去官网下载对应的包
比如：torch-1.8.0+cu111-cp36-cp36m-win_amd64.whl
torchvision-0.9.0+cu111-cp36-cp36m-win_amd64.whl

我的计算机是rtx3060的显卡，对应的duda版本是11.1（cu111），python36对应cp36, 64位操作系统

下载到本地后：进入conda环境激活对应的python36
conda env list
activate python36

pip install+ whl文件位置
比如：pip install D:\下载（电影，图片）\torch-1.8.0+cu111-cp36-cp36m-win_amd64.whl

（2）在线安装

进入conda环境激活对应的python36，
pip install torchvision
pip install torch
一般torch这个包比较大，从官网下载速度特别慢，还时不时地中断。

还可以指定国内镜像地址
例如： pip install --index-url https://pypi.douban.com/simple
比如: pip install --index-url https://pypi.douban.com/simple torchvison

但是，在线安装不指定具体的包名，一般会安装CPU版本的torch，torchvision，所以不推荐在线安装。

下附国内镜像常用地址
阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

各位大佬，看到觉得有用给个赞，嘿嘿！！！