1.conda创建虚拟环境

conda create -n phot37 python=3.7

phot37是环境名

2. 安装opencv-python

选择的版本是4.2.0,从网上查opencv有哪些版本,选择4.2.0.34,选择豆瓣源,豆瓣源还是挺快的

pip install opencv-python -i https://pypi.douban.com/simple

3.安装pytorch

pytorch官网选择对应CUDA版本的pytorch,安装即可,需要换源,从官网安装太慢了.

4.安装cuda

因为本机已经安装cuda11.4,但要求cuda10.2版本,因此首先把环境切换到虚拟环境中,再安装cudatoolkit=10.2

conda install cudatoolkit=10.2

或者

conda install cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

5.安装cudnn

conda install cudnn=8.1.1

或者

conda install cudnn=8.1.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

或者

conda install cudnn

应采用前两种方法,第三种不写版本,会自动适配,但是安全起见,写上版本再安装!

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐