手动安装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-2allfirst_rank_ecpm_v1~pc_rank_34-3-125225786-null-null.142v32pc_rank_34,185v2control&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
pip install torchvision-0.9.0+cu111-cp37-cp37m-win_amd64.whl

注意:根据自己下载的文件名更改就行了。
然后激活该环境: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

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

输入: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 

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

你可能感兴趣的:(目标检测,深度学习,opencv,人工智能,python,深度学习)