环境搭建02-Ubuntu16.04 安装CUDA和CUDNN、CUDA多版本替换

1、CUDA安装

(1)下载需要的CUDA版本
https://developer.nvidia.com/cuda-toolkit-archive

(2)安装

sudo sh cuda_8.0.61_375.26_linux.run

(3)添加环境

gedit ~/.bashrc

在文件末尾添加:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64
export PATH=$PATH:/usr/local/cuda-8.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-8.0

(4)使文件有效

source ~/.bashrc

(5)查看安装是否成功

cat /usr/local/cuda-8.0/version.txt

或者

nvcc -V   ## nvcc -version
2、CUDNN安装

(1)下载CUDNN
https://developer.nvidia.com/cudnn
需要注册用户,然后登录下载。注意cudnn的版本要和cuda的版本对应。

(2)安装

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*

(3)查看是否安装成功

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

以上步骤1和步骤2就是但版本CUDA的安装

3、CUDA多版本替换

例如:将CUDA-8.0更换为CUDA-10.0。这里,CUDA-8.0和CUDA-10.0以正确安装完成。可以通过创建cuda软链接指向不同版本的cuda版本。
首先,修改CUDA的路径,

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


这里cuda软链接指向cuda-8.0版本。
然后,删除指向cuda-8.0的软链接cuda,重新创建一个指向cuda-10.0版本的软链接。

rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
nvcc --version
  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您好!要在Ubuntu 16.04上安装CUDA,您可以按照以下步骤进行操作: 1. 首先,确保您的系统满足CUDA的要求。您可以访问NVIDIA的官方文档来查看支持的操作系统版本和硬件要求。 2. 下载CUDA Toolkit。您可以在NVIDIA的官方网站上找到CUDA Toolkit的下载链接。选择适合您系统的版本,并下载相应的.run文件。 3. 在终端中导航到您下载的.run文件所在的目录,并通过以下命令将其设置为可执行文件: ``` chmod +x cuda_*.run ``` 4. 运行.run文件以启动安装程序,通过按照提示进行安装。在安装过程中,您可能需要接受许可协议、选择安装选项和提供管理员密码等。 5. 安装完成后,需要将CUDA的路径添加到系统环境变量中。打开终端并编辑 `~/.bashrc` 文件: ``` nano ~/.bashrc ``` 6. 在文件末尾添加以下行来设置CUDA相关的环境变量: ``` export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 7. 保存并关闭文件,然后运行以下命令使环境变量生效: ``` source ~/.bashrc ``` 8. 最后,您可以验证CUDA是否成功安装。在终端中运行以下命令来检查CUDA版本: ``` nvcc --version ``` 如果安装成功,您应该能够看到CUDA版本信息。 这些是在Ubuntu 16.04上安装CUDA的基本步骤。请注意,具体的安装过程可能因您的系统配置而有所不同。如果遇到任何问题,议参考NVIDIA的官方文档或寻求社区支持。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值