阅读readme
阅读github上的yolov5的readme的环境配置,举例来说,yolov5_5.0要求如下。
删除原有出错误的环境
conda remove -n env_name --all -y
创建虚拟环境并指定python版本
conda create -n env_name python=3.9
安装pytorch
查看显卡驱动
nvidia-smi
下载压缩包
根据pytorch,python,CUDA Version版本的需求,从清华大学开源软件镜像站(Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror)选择对应的pytorch压缩包。
将下载的pytorch压缩包解压的虚拟环境中
命令:
conda install C:\Users\w\Downloads\pytorch-1.7.0-py3.8_cuda102_cudnn7_0.tar.bz2
install后面接的是压缩包的路径。
查看torch版本
虚拟环境中输入:
conda list torch
显示:
安装pytorch配套的其他库
pytorch配套的库有torchvision,torchaudio,cudatoolkit,去官网(Previous PyTorch Versions | PyTorch)查看配套库与pytorch搭配的版本,如图:
复制命令,但注意删除pytorch==1.7.0,因为pytorch之前已经安装过。得到如下的命令:
conda install torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
行文至此,pytorch的安装就结束啦,希望读到这篇文章的有缘人能够顺利安装pytoch!
环境的变化
在安装yolov5_5.0的requirement.txt,安装文件将原有的torch1.7及附属库给删除了,并安装了新的库,如下:
但不用担心,程序能够正常的运行。
虚拟环境重新命名
conda rename -n old_name new_name