安装 pytorch-gpu 版本时,首先得查看本机的 cuda 版本。
win+R 打开 cmd,输入 nvcc -V 即可得到版本号(根据版本号选择对应的 pytorch-gpu 版本),本机 cuda 版本号为11.1。
然后复制该网址 https://download.pytorch.org/whl/torch_stable.html
去找对应的 pytorch-gpu 版本。玫红色为 cuda 版本号(检索到自己对应的 cuda 版本号),黑色为 torch 版本号,灰色为 python 版本号,后面为选择是 win 系统还是 linux 系统。全部选择完毕单击下载,还需要选择下载对应的
选择下载对应的 torchvision (在以上同一网页中下载)
下载后将两个文件复制到对应的环境的 Scripts 中,例如
在cmd中依次输入环境路径以及激活环境
然后 pip install 安装即可!
如何检查是否安装成功呢??
打印 torch.cuda.is_available(),返回 ture 则说明安装成功!!!
安装过程中遇到的问题!!!
若遇到以下问题,仔细检查当前环境所对应的 python 版本与下载的版本是否匹配。在安装 pytorch 和 torchvision 时,两者的版本号也要对应。
安装 opencv,只用执行以下两个命令就可以简单的进行安装,这边选择了3.4.15 的原因是在 3.4.2 以上存在专利版权问题,比如特征提取等功能无法正常使用。
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn opencv-python==3.4.1.15
安装过程中遇到的问题!!!
若遇到以下问题,是安装的 opencv 版本与 python 版本不匹配原因造成的。换成对应版本即可
若以下安装报错:
pip install opencv-python==3.4.1.15 https://pypi.tuna.tsinghua.edu.cn/simple/
则换成:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn opencv-python==3.4.1.15
总而言之,需要 python,cuda,pytorch,opencv 之间的版本一一对应,这样安装就会一路畅通~