CUDA, cuDNN, Tensorflow,Mask RCNN配置实战
因为之前环境配置失败,一怒之下刷机推倒重来,跑过来记录这次环境配置的经历。参考了站内站外许多教程,整理如下。
- CUDA
- cuDNN
- Anaconda
- Tensorflow1.5.0
- Keras2.1.6
- Mask RCNN 案例
- Memory: 32GB
- Processor: Intel i9-10980 @ 2.4GHZ x 16
- Graphics: GeForce RTX 2070 Super with Max-Q Design/PCIe/SSE2
- System: Ubuntu 22.04
-
CUDA的安装
因为不是第一次安装CUDA,所以这部分写的不是非常详细,这里只记录下本人的安装流程,详细安装教程可以参考这里。安装之前,首先对显卡的驱动进行确认:
nvidia-smi
如果输出结果为:
证明显卡驱动已经安装成功。未配置driver的话,会出现“Make sure that the latest NVIDIA driver is installed and running.” 这时候需要使用如下命令:sudo ubuntu-drivers autoinstall
如果报错的话,需要参考如何禁用nouveau,重启,然后输入以下命令验证是否生效
lsmod | grep nouveau
没有输出结果,则表明禁用成功。然后就可以开始CUDA的安装。我选择安装CUDA11.2,大家可以在网上选择正确版本进行安装。下面是我选择的安装选项。
确定之后,在选好的路径下执行wget https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.run sudo sh cuda_11.2.1_460.32.03_linux.run
在安装的时候,会