Ubuntu18.04 cuda10.1安装与卸载,NVIDIA驱动安装

本文详细指导了从卸载CUDA 10.0,通过`sudo ubuntu-drivers`安装NVIDIA驱动,遇到问题后的解决方案,再到CUDA 10.1的全新安装,并配置环境变量。涉及步骤包括cuda10.0卸载、nvidia驱动自动修复及cudnn和torch的安装。
摘要由CSDN通过智能技术生成

Ubuntu18.04 cuda10.1安装与卸载,NVIDIA驱动安装

1 cuda 10.0卸载

sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl 

在这里插入图片描述
删除残余

sudo /usr/local/cuda

卸载完成!

2 安装nvidia驱动

sudo ubuntu-drivers devices

在这里插入图片描述

sudo ubuntu-drivers autoinstall

在这里插入图片描述
但是安装完后重启系统进不去
这里可以进入advanced options模式
选择recovery模式,选择dpkg进行自动修复,需要联网重启,即可

在这里插入图片描述

3安装cuda 10.1

chmod +x cuda_10.1.168_418.67_linux.run
sudo ./cuda_10.1.168_418.67_linux.run 

驱动不选择
在这里插入图片描述
添加环境变量

sudo gedit ~/.bashrc
export CUDA_HOME=/usr/local/cuda 
export PATH=$PATH:$CUDA_HOME/bin 
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存退出

source ~/.bashrc

验证是否成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

接下来解压cudnn
用下列命令复制cudnn目录

sudo cp cudnn-10.1-linux-x64-v7.6.5.32/cuda/lib64/* /usr/local/cuda-10.1/lib64/
sudo cp cudnn-10.1-linux-x64-v7.6.5.32/cuda/include/* /usr/local/cuda-10.1/include/

查看版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

在这里插入图片描述

4 Anconda 创建环境出错

在这里插入图片描述
改成 python==3.8就可以了

5 安装torch

激活环境后

pip install torch-1.8.0+cu101-cp38-cp38-linux_x86_64.whl
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值