我想在自己的机器上安装ubuntu18.04系统,硬件上安装了NVIDIA 1080Ti,软件我想安装11.2版本的cuda和8.5版本的cudnn,请帮我写一份安装的脚本
时间: 2024-06-08 19:12:50 浏览: 55
ubuntu安装显卡驱动+cuda+cudnn教程(18.04+cuda10+cudnn9.05为例)
5星 · 资源好评率100%
好的,以下是安装脚本:
```bash
# 安装CUDA 11.2
wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run
sudo sh cuda_11.2.0_460.27.04_linux.run --silent --toolkit --override
echo 'export PATH=/usr/local/cuda-11.2/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
# 安装cuDNN 8.5
wget https://developer.download.nvidia.com/compute/redist/cudnn/v8.2.2/cudnn-11.2-linux-x64-v8.2.2.26.tgz
tar -xzvf cudnn-11.2-linux-x64-v8.2.2.26.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.2/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.2/lib64/
sudo chmod a+r /usr/local/cuda-11.2/include/cudnn*.h /usr/local/cuda-11.2/lib64/libcudnn*
# 检查安装是否成功
nvcc -V
```
运行以上脚本后,CUDA和cuDNN应该已经成功安装在您的机器上了。
阅读全文