全站最帅😎
发布于 2022-07-19 / 1331 阅读
0
0

安装 pytorch-gpu,opencv。

安装 pytorch-gpu 版本时,首先得查看本机的 cuda 版本。
win+R 打开 cmd,输入 nvcc -V 即可得到版本号(根据版本号选择对应的 pytorch-gpu 版本),本机 cuda 版本号为11.1。
image.png

然后复制该网址 https://download.pytorch.org/whl/torch_stable.html
去找对应的 pytorch-gpu 版本。玫红色为 cuda 版本号(检索到自己对应的 cuda 版本号),黑色为 torch 版本号,灰色为 python 版本号,后面为选择是 win 系统还是 linux 系统。全部选择完毕单击下载,还需要选择下载对应的
image.png

选择下载对应的 torchvision (在以上同一网页中下载)
image.png

下载后将两个文件复制到对应的环境的 Scripts 中,例如
image.png

在cmd中依次输入环境路径以及激活环境
image.png

然后 pip install 安装即可!
image.png

如何检查是否安装成功呢??
打印 torch.cuda.is_available(),返回 ture 则说明安装成功!!!

安装过程中遇到的问题!!!
若遇到以下问题,仔细检查当前环境所对应的 python 版本与下载的版本是否匹配。在安装 pytorch 和 torchvision 时,两者的版本号也要对应。
image.png

安装 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
image.png
image.png

安装过程中遇到的问题!!!
若遇到以下问题,是安装的 opencv 版本与 python 版本不匹配原因造成的。换成对应版本即可
image.png
若以下安装报错:
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
image.png

总而言之,需要 python,cuda,pytorch,opencv 之间的版本一一对应,这样安装就会一路畅通~


评论