Ubuntu16安装cuda和cudnn的一点问题及解决方法

Ubuntu16安装cuda和cudnn的一点问题及解决方法


ubuntu 16.04默认安装了第三方开源的驱动程序nouveau,安装nvidia显卡驱动首先需要禁用nouveau,不然会碰到冲突的问题,导致无法安装nvidia显卡驱动。

sudo gedit /etc/modprobe.d/blacklist.conf ->打开此文件。
blacklist nouveau->在文件结尾输入,保存并退出。
sudo update-initramfs -u ->执行,让修改后的文件生效。
重启电脑
lsmod | grep nouveau ->测试nouveau是否禁用成功,是的话,执行命令后应该什么都没出现。
https://blog.csdn.net/qq_42730894/article/details/107655026 ->之后的操作。

安装cuda10.1:
下载驱动支持对应的.run文件。
放到home文件夹下。
sudo sh xxx.run ->执行
在这里插入图片描述
按提示,添加环境变量。sudo gedit /etc/profile
在文件最后添加:
export PATH=/usr/local/cuda-10.1/binKaTeX parse error: Expected '}', got 'EOF' at end of input: {PATH:+:{PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64KaTeX parse error: Expected '}', got 'EOF' at end of input: …LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
重启电脑。
nvcc -V验证cuda是否安装成功。

安装cudnn 7.6.3 for cuda10.1
下载cudnn for linux.tar。
解压出一个cuda文件夹(tar -xzvf cudnn-9.0-linux-x64-v7.tgz)。
安装就是把解压出的文件复制到指定位置,并给予权限。
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
测试cudnn是否可用:
from torch.backends import cudnn
print(cudnn.is_available())

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要在Ubuntu安装CUDACuDNN,您可以按照以下步骤进行操作: 1. 安装CUDA: a. 访问NVIDIA开发者网站(https://developer.nvidia.com/cuda-downloads)并下载适用于您的Ubuntu版本的CUDA安装包。 b. 打开终端,并切换到您下载的CUDA安装包所在的目录。 c. 运行以下命令解压安装包: ``` sudo sh cuda_*.run ``` d. 按照安装程序的指示进行安装。在安装过程中,您可以选择是否安装NVIDIA驱动程序和CUDA Toolkit。 2. 配置CUDA环境变量: a. 打开终端,并打开`~/.bashrc`文件: ``` gedit ~/.bashrc ``` b. 在文件末尾添加以下行: ``` export PATH=/usr/local/cuda-<version>/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 注意将`<version>`替换为您所安装CUDA版本号。 c. 保存并关闭文件。 d. 运行以下命令使更改生效: ``` source ~/.bashrc ``` 3. 安装CuDNN: a. 访问NVIDIA开发者网站(https://developer.nvidia.com/rdp/cudnn-archive)并下载与您所安装CUDA版本兼容的CuDNN库文件。 b. 解压下载的CuDNN库文件。 c. 打开终端,并切换到解压后的CuDNN目录。 d. 运行以下命令将CuDNN库文件复制到CUDA安装目录: ``` sudo cp cuda/include/cudnn.h /usr/local/cuda-<version>/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-<version>/lib64 ``` 注意将`<version>`替换为您所安装CUDA版本号。 现在,您应该已成功安装和配置了CUDACuDNN
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值