windows系统下pytorch搭建GPU环境，通用

用pytorch搭建GPU环境，通用

一 ：查找cuda和cudnn

如何查看版本， 命令行：nvidia-smi ；
方法链接 复制替换
官方驱动 下载链接

cuda cudnn 显卡 三者相匹配，记住cuda的版本号 如cuda 11.1

cuda	cudnn

上面一步会耗点时间，可以查看上面链接

二 ：选择 python 版本

用anaconda搭建好自己的环境(这步我省略了。。。)

如3.8，高一点新一点，torch中一些更新只对高版本py开放，不低于3.7。
3.6就出现了某些不支持的操作

三 ：选择 对应 torch 版本

1. 在线安装

同时在网上 查找对应版本pytorch 找对应的（根据注释中的Cuda版本11.1） Ctrl+F

确定版本为 pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
其中开头分别是： cu111/torch-1.9.1 cu111/torchvision-0.10.1 找对应得python版本，此时为3.8cu111/torchaudio 后面这个是音频 ，可不安装

2. 本地安装

• 由于在线安装基本会因网络波动导致失败，选择离线安装

对应版本查找 https://download.pytorch.org/whl/torch_stable.html

Ctrl+F ➡数字是对应的版本

cu111/torch-1.9.1%2Bcu111-cp38-cp38-win_amd64	cu111/torchvision-0.10.1%2Bcu111-cp38-cp38-win_amd64

四 ：下载完毕，打开环境中prompt，

切换到下载完成的文件夹：

1）切换到所在盘符字母：F ；F

2）cd 具体目录：D:\360极速浏览器下载 ； cd D:\360极速浏览器下载

pip install torch-1.9.1+cu111-cp38-cp38-win_amd64.whl

pip install torchvision-0.10.1+cu111-cp38-cp38-win_amd64.whl

出现了下载别的包，是依赖包，不用惊讶！

个人平常会在pycharm该导入环境，然后使用pycharm里面终端进行包的安装与卸载

五 ： 验证

输入python，别输入py两个字母！！如果出现了不是该环境的python版本，去看看系统环境变量中是不是加入了默认base中python的路径！！！

六 ： 接下来开启CV阶段

今天预计划学习商汤的 mmdetection使用（这个环境是独立的）！
然而被喊过去调试别的项目，环境没有新建。。。。
这就是此篇文章的由来了！！

有安装问题欢迎留言讨论！