目录

 

一、创建Pytorch环境

1.创建虚拟环境

2.安装对应版本的pytorch

二、安装opencv

一、创建Pytorch环境

1.创建虚拟环境

        创建虚拟环境conda create -n 环境名字(英文) python=x.x(python版本),如下,我就是创建了一个名字叫pytorch,python是3.8版本的环境。

conda create -n pytorch python=3.8

        在base环境中执行如上的命令,就会创建一个新的虚拟环境,这个虚拟环境会安装一些基础的包,如下图所示。询问是否安装的时候,输入y。就可以创建环境了。

conda当前环境pytorch版本 conda创建pytorch环境_conda当前环境pytorch版本

        执行如下命令,激活这个环境。conda activate 虚拟环境名称。

conda activate pytorch

conda当前环境pytorch版本 conda创建pytorch环境_python_02

        发现base环境变成了pytorch

2.安装对应版本的pytorch

        首先想到的肯定是官网安装。conda+重新开放的清华源=神速。此处是我自己的一个大坑。按照大量帖子教学,打开pytorch的官网(https://pytorch.org/)。

        由于开头我们通过驱动检测到我的显卡为 RTX2060,最高支持cuda11.1版本,所以我们选择cuda11.1版本的cuda。

conda当前环境pytorch版本 conda创建pytorch环境_python_03

        然后将蓝色高亮的内容复制下来到cmd下执行, 执行后即安装成功。再进入python检验一下导包貌似没有问题。

conda当前环境pytorch版本 conda创建pytorch环境_虚拟环境_04

        但是在后面跑代码的时候,很意外地碰到了一个错误提示AssertionError:Torch not compiled with CUDA enabled,如下图所示:

conda当前环境pytorch版本 conda创建pytorch环境_官网_05

        问题的发现可以在终端中输入python命令,运行python后,输入

import torch 
print(torch.__version__) 
print(torch.cuda.is_available())

        如果输出的结果是False,那么说明当前的Pytorch版本无法使用显卡。

解决办法:

        看了几篇帖子才知道,cuda是向下兼容的,要确保新建的环境的cuda版本要比本地的基础环境低!!

        偶然发现,使用官网conda命令下载的,居然全是CPU版本。conda为什么只能下cpu版本,可能是清华源没有对应cuda版本的pytorch,所以会装cpu版的。

conda当前环境pytorch版本 conda创建pytorch环境_python_06

开始解决

1、卸载pytorch

pip uninstall torch

2.尝试用pip方法重新安装

        在此处(https://download.pytorch.org/whl/torch_stable.html)寻找符合自己版本的pytorch-cuda版本,torchvision版本下载。然后进入cmd,使用cd进入whl文件所在的目录,输入

pip install torch-1.7.0+cu110-cp38-cp38-win_amd64.whl

pip install torchvision-0.8.1+cu110-cp38-cp38-win_amd64.whl

        安装成功后,测试环境报错ImportError: numpy.core.multiarray failed to import。通过网上查找资料发现一般情况都是卸载numpy重新安装就好了,但是我卸载了几次重新安装还都是报该错误。

解决步骤:

        在https://www.lfd.uci.edu/~gohlke/pythonlibs/# numpy依据自己安装的python版本下载对应版本的numpy+mkl至本地安装

conda当前环境pytorch版本 conda创建pytorch环境_官网_07

        安装完再测试成功!

conda当前环境pytorch版本 conda创建pytorch环境_虚拟环境_08

二、安装opencv

进入Anaconda Prompt (Anaconda3)

conda当前环境pytorch版本 conda创建pytorch环境_python_09

输入python即可查看python,比如我的是3.8.8版本。

conda当前环境pytorch版本 conda创建pytorch环境_python_10

        进入网页https://www.lfd.uci.edu/~gohlke/pythonlibs/下载OpenCv的whl文件
python版本是多少就选带cp多少的,我的是3.8.8版本,我就选带cp38的,电脑是64位就选带win64的
我用的是opencv_python-4.5.5--cp38-win_amd64.whl,然后进入cmd,使用cd进入whl文件所在的目录,输入

pip install opencv_python-4.5.5--cp38-win_amd64.whl

conda当前环境pytorch版本 conda创建pytorch环境_python_11

        等待安装,用import cv2测试

conda当前环境pytorch版本 conda创建pytorch环境_conda当前环境pytorch版本_12