Ubuntu安装和卸载CUDA和CUDNN的实现
在Ubuntu系统中,安装和卸载CUDA以及CUDNN是深度学习和GPU计算的重要环节。CUDA是NVIDIA提供的用于加速计算的应用程序编程接口(API),而CUDNN是CUDA的深度神经网络库,优化了卷积神经网络等计算密集型任务。以下是详细的操作步骤: 我们来讲解如何在Ubuntu上安装CUDA和CUDNN。 1. **安装显卡驱动**: 在安装CUDA之前,需要确保正确安装了NVIDIA显卡驱动。需要禁用内建的开源驱动nouveau。编辑`/etc/modprobe.d/blacklist.conf`文件,添加如下内容: ``` blacklist nouveau options nouveau modeset=0 ``` 更新初始化ramfs: ``` sudo update-initramfs -u ``` 重启系统,检查nouveau是否被禁用,如果没有`nouveau`的输出,则已成功。从NVIDIA官方网站下载适合你显卡的驱动,例如对于RTX2070,下载名为`NVIDIA-Linux-x86_64-410.93.run`的文件。然后,执行以下命令卸载旧驱动,安装新驱动并重启: ``` Ctrl-Alt-F1 sudo service lightdm stop sudo apt-get remove --purge nvidia* sudo chmod +x NVIDIA-Linux-x86_64-410.93.run sudo ./NVIDIA-Linux-x86_64-410.93.run --uninstall sudo ./NVIDIA-Linux-x86_64-410.93.run sudo service lightdm start reboot ``` 2. **卸载CUDA**: 卸载CUDA可以通过执行CUDA自带的卸载脚本来完成。例如,对于CUDA 8.0,运行: ``` sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl ``` 随后删除遗留的CUDA目录: ``` sudo rm -rf /usr/local/cuda-8.0/ ``` 3. **安装CUDA**: 从NVIDIA官网下载CUDA的.run文件,例如CUDA 10.0的安装包。给文件添加执行权限并运行: ``` chmod +x cuda_10.0.130_410.48_linux.run ./cuda_10.0.130_410.48_linux.run ``` 安装过程中,选择接受协议,不安装驱动(已安装),并确认安装CUDA工具包。 4. **安装CUDNN**: CUDNN通常与CUDA版本配套使用。下载CUDNN对应的版本(如CUDNN 7.4.2),解压并复制到CUDA目录: ``` tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.tgz sudo cp -P include/* /usr/local/cuda/include/ sudo cp -P lib64/* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* ``` 5. **配置环境变量**: 添加CUDA和CUDNN到环境变量中,编辑`~/.bashrc`: ``` nano ~/.bashrc ``` 添加以下行: ``` export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 保存并关闭文件,然后执行: ``` source ~/.bashrc ``` 6. **验证安装**: 可以通过运行`nvidia-smi`和`nvcc --version`命令来检查CUDA和NVIDIA驱动是否正确安装,以及CUDA编译器的版本。 卸载CUDA和CUDNN时,只需按照上述步骤逆向操作即可。务必谨慎操作,避免影响到其他依赖CUDA的软件或系统功能。在进行这些操作时,确保你了解每个步骤的含义,以防止可能出现的问题。