Ubuntu20.04 LTS CUDA安装、TF-GPU、Torch-GPU

最新推荐文章于 2025-07-11 16:52:32 发布

拒绝气泡

最新推荐文章于 2025-07-11 16:52:32 发布

阅读量420

点赞数

CC 4.0 BY-SA版权

分类专栏：机器和深度文章标签： TF-GPU GPU环境安装 pytorch-GPU

本文链接：https://blog.csdn.net/CSDNgaoqingrui/article/details/119414586

本文详细介绍了如何在Ubuntu20.04 LTS上卸载及安装NVIDIA驱动、CUDA和cuDNN，以及如何配置和测试TensorFlow-GPU和PyTorch-GPU环境。步骤包括：安全卸载NVIDIA驱动，查找并安装合适驱动，安装CUDA和cuDNN，以及通过conda和pip安装TF2-GPU和pytorch-gpu。最后，提供了测试GPU是否正常工作的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

CUDA 卸载没有 uninstall

卸载办法

# 用 runfile 方式安装的删除方法：
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl

# 用 package manager 方式安装的删除方法：
sudo apt-get --purge remove cuda-8.0
sudo apt autoremove
sudo apt-get autoclean
# cudnn文件和samples残留在 /usr/local/cuda-8.0/，删除
sudo rm -rf /usr/local/cuda-8.0/

卸载驱动

sudo apt-get remove --purge nvidia*

查找可安装的驱动

ubuntu-drivers devices

安装推荐驱动

sudo apt install nvidia-driver-460

查看NVIDIA驱动版本

cat /proc/driver/nvidia/version
驱动和cudaToolkit版本对应
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
![image.png](https://img-blog.csdnimg.cn/img_convert/c640932a8915f10b1b4aef12fd7667d0.png#align=left&display=inline&a