目标:在Ubuntu 18.04配置CUDA 11.3 和 cuDNN 8.8.0
环境说明:容器内已配置显卡驱动+torch-gpu版
- 检查容器发现没有安装CUDA工具包:
nvcc -V
- 检查当前显卡驱动信息及其支持的最高CUDA版本:
nvidia-smi
- 安装CUDA:
- 从nVidia官网下载对应版本的CUDA
- 选择
runfile(local)
版本后,官网会直接给出了安装指令
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_