2. 运行下面的命令进行安装:
chmod x cuda_11.0.2_450.51.05_linux.run
sudo sh ./cuda_11.0.2_450.51.05_linux.run
可能会报一个警告:
前面已经卸载过旧版本了直接Continue就好。然后根据提示选择安装选项,注意不要勾选第一个安装显卡驱动的,因为之前已经安装过了。安装完成后提示
3. 根据上图提示需要配置环境变量:
nano ~/.bashrc
再文件最后加入以下语句:
export CUDA_HOME=/usr/local/cuda-11.0
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
然后使其生效:
source ~/.bashrc
3. 可以使用命令nvcc -V查看安装的版本信息:
也可以编译一个程序测试安装是否成功,执行以下几条命令:
cd ~/Softwares/cuda/NVIDIA_CUDA-11.0_Samples/1_Utilities/deviceQuery
make
./deviceQuery
正常的话会有相应输出,打印显卡的信息。
3.3 安装CuDNN进入到CUDNN的下载官网:cuDNN Download | NVIDIA Developer(https://developer.nvidia.com/rdp/cudnn-download),然点击Download开始选择下载版本,当然在下载之前还有登录,选择版本界面如下: