deepin20.6深度学习环境配置(二)--miniconda、cuda、cudnn、pip换源

材料:Ubuntu、显卡

一、安装Miniconda3

1.下载地址Miniconda3-latest-Linux-x86_64.sh

# 安装命令
./Miniconda3-latest-Linux-x86_64.sh

注意:如果在vm虚拟机里面安装,可能会出现如下问题:
在这里插入图片描述
解决方法:在"END"处,键盘输入“yes”按回车,就会出现下面的情况:
在这里插入图片描述

#继续执行
vim ~/.bashrc
#在最后添加如下,并保存退出。
export PATH=/home/用户名/miniconda3/bin:$PATH
#环境变量生效
source ~/.bashrc

2.进入miniconda3

cd /home/用户名/miniconda3/bin

3.创建虚拟环境

# 环境名yolov5_env,python版本3.8
conda create -n yolov5_env python=3.8

4.激活虚拟环境

conda activate yolov5_env

激活前后

5.conda换源参考

二、安装cuda、cudnn

cuda下载,30系列显卡前选择10.2,30系列显卡选择11.1.1
cudnn下载(需注册登录),根据cuda版本选择对应的cudnn(推荐cuda10.2+cudnn7.6.5、cuda11.1+cudnn8.1.1)
在这里插入图片描述

安装:

sudo sh cuda_10.2.89_440.33.01_linux.run

在这里插入图片描述

添加环境变量:

gedit ~/.bashrc

添加:

export PATH=/usr/local/cuda-10.2/bin${
    
    PATH:+:${
    
    PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新环境变量配置:

source ~/.bashrc

验证:

nvcc -V

在这里插入图片描述
解压cudnn:

tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz

复制文件,并修改权限:

sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
# 顺序一定按照以下顺序
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb

验证cudnn:

# 将cuDNN示例复制到可写路径
$ cp -r /usr/src/cudnn_samples_v7/ $HOME
# 进入到可写路径
$ cd  $HOME/cudnn_samples_v7/mnistCUDNN
# 编译mnistCUDNN示例
$ make clean && make
# 运行mnistCUDNN示例
$ ./mnistCUDNN

运行结果出现即可:

Test passed!

三、pip换源参考:

猜你喜欢

转载自blog.csdn.net/wave789/article/details/126170817