材料: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!