手动安装torch和torchviosn教程

一、根据你笔记本显卡的算力和驱动，你安装的GPU版本为：
NVIDIA 驱动下载：https://www.nvidia.cn/Download/index.aspx?lang=cn
cuda下载链接：https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
cudnn下载地址：https://developer.nvidia.com/cudnn
cuda和cudnn版本匹配：https://blog.csdn.net/tangjiahao10/article/details/125225786?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522165776217516782391827998%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=165776217516782391827998&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2_allfirst_rank_ecpm_v1~pc_rank_34-3-125225786-null-null.142^v32pc_rank_34,185^v2control&utm_term=%E6%9F%A5%E7%9C%8Bcuda%E5%92%8Ccudnn%E7%89%88%E6%9C%AC&spm=1018.2226.3001.4187
举例：
cuda 10.2
cudnn 7.6.5

二、从github下载到了一个项目后，第一件事就是查看requirements.txt，里头的torch和torchvision的版本，
，若在里头已经指定了python版本，例如python=3.7的版本，因为你网速不太好，所以我建议你离线安装，
首先进入这个网址：https://download.pytorch.org/whl/torch_stable.html
选择cuda对应的torch和torchvision的版本，
例如：torchvision-0.9.0+cu111-cp37-cp37m-win_amd64.whl
torchvision-0.9.0：torchvision 版本号
cu111： cuda版本
win_amd64：windows版本
cp37：python=3.7
同理可得：torch-1.8.0+cu111-cp37-cp37m-win_amd64.whl，版本号一一对应。
注意：cuda10.2的可能找不到，可以选择10.0的，建议下载好后，把这个包放置于与项目同一个文件夹中。

三、在conda终端（一定要是这个：commend prompt）输入以下命令，新建python环境。
conda create --name 环境的名称(要求英文，例如mypytorch，总之随便取) python=python的版本，例如3.6，3.7，3.8，3.9
举例：conda create --name yolov5 python=3.7
创建了一个名为yolov5，python版本为3.7的环境，
然后手动安装：

pip install torch-1.8.0+cu111-cp37-cp37m-win_amd64.whl
1

pip install torchvision-0.9.0+cu111-cp37-cp37m-win_amd64.whl
1

注意：根据自己下载的文件名更改就行了。
然后激活该环境：conda activate yolov5
激活后，前面会显示你的当前环境是什么。

四、根据requirements.txt配置其他环境。
在输入以下命令之前记得将刚刚那两个安装好的torch和torchvision注释掉，例如：
打开requirements.txt后，

matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.1
Pillow>=7.1.2
PyYAML>=5.3.1
requests>=2.23.0
scipy>=1.4.1  # Google Colab version
#torch>=1.7.0,!=1.12.0  # https://github.com/ultralytics/yolov5/issues/8395
#torchvision>=0.8.1,!=0.13.0 # https://github.com/ultralytics/yolov5/issues/8395
tqdm>=4.41.0
1
2
3
4
5
6
7
8
9
10

保存，注释的目的是防止重复安装。

输入：pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
安装所有其他环境，
这样恭喜你，不报错的话，环境基本配置好了。

5、关于加国内镜像源
例如：pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
这个是国内的网址：https://pypi.tuna.tsinghua.edu.cn/simple
加上这个，安装下载的速度飙升。

*除了一次性安装所有包，你还可以单独安装一个任意版本的包，例如安装：numpy==1.18.5

pip install -i numpy==1.18.5 https://pypi.tuna.tsinghua.edu.cn/simple 
1

*关于卸载某个包
直接输入命令：pip uninstall numpy
不管什么版本都可以卸载。