安装nvcc & CUDNN

一、安装NVCC

1、去官网下载一个run包https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=CentOS&target_version=7&target_type=runfilelocal
我下的这个,
在这里插入图片描述

2、安装~~~

sudo sh cuda_*.run

3、有几个选项
第一个ACCEPT接受
第二个选择安装的东西
带X的是要安装的,有驱动就把驱动关掉,只安装cuda tool kit即可
然后移动到最下面install
点回车
等待片刻安装好了
4、环境变量

vim ~/.bashrc
#下面加进去有
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
#source一下
source ~/.bashrc

nvcc -V检查安装是否成功,如下图则表示安装成功。
在这里插入图片描述
二、安装CUDNN

在这里插入图片描述

  • 4
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
安装 CUDA 和 cuDNN 是进行深度学习的必备步骤。下面是在 Ubuntu 上安装 CUDA 和 cuDNN 的步骤: ### 安装 CUDA 1. 下载 CUDA:访问官网 `https://developer.nvidia.com/cuda-downloads`,选择您的操作系统、体系结构和 CUDA 版本,下载 CUDA。 2. 安装 CUDA:运行以下命令进行安装: ```bash sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb sudo apt-get update sudo apt-get install cuda ``` 3. 设置环境变量:将以下内容添加到 `~/.bashrc` 文件中: ```bash export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 4. 重新加载 `.bashrc` 文件:运行以下命令重新加载 `.bashrc` 文件: ```bash source ~/.bashrc ``` 5. 验证 CUDA 安装:运行以下命令以确认 CUDA 是否正确安装: ```bash nvcc -V ``` 如果您看到 CUDA 版本信息,则说明 CUDA 安装成功。 ### 安装 cuDNN 1. 下载 cuDNN:访问官网 `https://developer.nvidia.com/rdp/cudnn-download`,选择您的操作系统、体系结构和 cuDNN 版本,下载 cuDNN。 2. 解压缩 cuDNN:运行以下命令解压缩 cuDNN: ```bash tar -xzvf cudnn-<version>.tgz ``` 3. 复制文件:运行以下命令将文件复制到 CUDA 的安装目录中: ```bash 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* ``` 注意,如果您的 CUDA 安装目录不是 `/usr/local/cuda`,则需要相应地更改路径。 4. 验证 cuDNN 安装:运行以下命令以确认 cuDNN 是否正确安装: ```bash cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 ``` 如果您看到 cuDNN 版本信息,则说明 cuDNN 安装成功。 希望这些步骤对您有所帮助。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值