ubuntu20.05+GTX1050Ti驱动+CUDA10.2+CUDNN8+pytorch最后OK

原创

已于 2022-04-15 07:55:34 修改 · 1k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习

于 2021-06-12 15:04:21 首次发布

该教程详细介绍了如何在Ubuntu系统中安装NVIDIA驱动、低版本GCC，以及CUDA和CuDNN。首先通过检查GPU、删除旧驱动、更新系统并安装推荐的NVIDIA驱动。接着安装GCC 7，设置为默认版本。然后，安装CUDA 10.2，不包含驱动，并在.bashrc中配置路径。再安装CuDNN，包括解压、复制文件到CUDA目录、安装deb包并验证。最后，安装PyTorch并测试CUDA可用性。

cankaole

https://www.jianshu.com/p/a16aaa1551d2

1/注意：安装驱动

（1）检查自己电脑的gpu是否CUDA-capable

lspci | grep -i nvidia

（2）删除旧的NVIDIA驱动

sudo apt-get remove nvidia-*
sudo apt-get autoremove

（3）

sudo apt-get update

（4）使用下面的命令查看系统推荐安装哪个版本的NVIDIA显卡驱动

ubuntu-drivers devices

（5）选择你看到推荐版本安装，本例使用nvidia-driver-440，然后安装几个必要组件，命令如下

sudo apt-get install nvidia-settings nvidia-driver-440 nvidia-prime

（6）

sudo reboot

（7）查看是否安装成功

nvidia-smi

2\安装低版本gcc

sudo apt install gcc-7  g++-7

将gcc 7 g++7 设置为优先级最高

sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 90 --slave /usr/bin/g++ g++ /usr/bin/g++-7 --slave /usr/bin/gcov gcov /usr/bin/gcov-7
sudo update-alternatives --install /usr/bin