这段时间在做YOLO v4的目标检测,由于对CUDA和CUDNN有要求,只能重新搭建环境了。
一、CUDA安装
1、安装多版本的CUDA,可自由切换
CUDA官网:https://developer.nvidia.com/cuda-toolkit-archive
然后,选择对应的安装包,选择runfile类型的安装文件,以便后面设置每个cuda的安装路径。
下载完成后,在命令行中进入cuda_10.0.130_410.48_linux.run文件所在的目录中,运行下面的指令进行安装:
sudo sh cuda_10.0.130_410.48_linux.run
安装过程中在建立软链接时需要注意一下,如果你是第一次安装cuda,那么毫无疑问输入y(yes),但是如果你是