Ubuntu21.04同时安装多个cuda

由于不同的项目需要,服务器需要安装多个不同版本的cuda方便程序运行,在此做个记录

本人电脑之前已经成功安装了cuda10.2的版本,现在需要新增一个cuda10.1的环境

如果还没安装任何cuda版本,请参考我的这篇链接进行cuda和cudnn的安装:

Ubuntu21.04安装cuda10.2和cudnn_Nudge636的博客-CSDN博客_cudnn安装教程

安装第二个cuda主要是参照了这篇博客:

Ubuntu Cuda版本切换_孤高丶逛侠的博客-CSDN博客_ubuntu 切换cuda版本

Ubuntu安装多版本cuda 多版本之间切换_wangyunpeng_bio的博客-CSDN博客_cuda多版本切换

安装步骤:

一、首先去cuda官方下载新安装的cuda版本:

CUDA Toolkit Archive | NVIDIA Developer

比如我要新安装cuda-10.1,就下载CUDA Toolkit 10.1,下载以后是一个.run文件

二、给予安装文件权限并安装

chmod +x cuda_10.1.105_418.39_linux.run 

./cuda_10.1.105_418.39_linux.run --override

安装的时候注意只用选Toolkit就可以,其他的应该不用装,注意不要选软连接

此时输入cd /usr/local,然后输入ls应该有三个文件,一个cuda,一个cuda10.2(原先安装的),一个cuda10.1

三、环境cuda配置

首先打开环境配置文件

sudo gedit ~/.bashrc

在文档的末尾添加这三行,cuda-10.1的地方修改为自己新安装的cuda版本

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

输入完后保存,退出bashrc后在终端输入

source ~/.bashrc

此时再输入nvcc-V,应该就能看到cuda版本成功切换为新安装的版本10.1了

四、下载cuda10.1配套的cudnn并安装

开头那篇博客里有说到,cuda是为了让程序能够并行计算,而cudnn是为了让gpu能跑深度学习的东西,所以安装了新的cuda10.1后,得安装配套的cudnn

首先进入cudnn的下载官网

cuDNN Archive | NVIDIA Developer

找到for cuda 10.1对应的cudnn,只用下载cuDNN Library for Linux(Power)就行

下载好后是一个tgz的压缩文件

先解压缩,然后参照下边的代码复制一下文件到cuda10.1的文件夹中就可以啦

tar -zvxf cudnn-10.1-linux-ppc64le-v8.0.5.39.tgz
sudo chmod 777 -R *
sudo cp cuda/targets/ppc64le-linux/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp cuda/targets/ppc64le-linux/lib/libcudnn* /usr/local/cuda-10.1/lib64

至此,新的cuda环境已经装好,嘿嘿~

你可能感兴趣的:(python,深度学习)