Ubuntu 多用户分别使用不同版本CUDA (2021亲测可行)

*****************

禁止私自转载

*****************

 

参考链接：

https://blog.csdn.net/qq_41202069/article/details/108286319

https://www.cnblogs.com/yhjoker/p/10972795.html

 

1. 下载想要的CUDA和对应版本CUDNN

https://developer.nvidia.com/cuda-toolkit-archive

https://developer.nvidia.com/cudnn

 

2. 如下图安装CUDA

* 额外版本CUDA不要安装显卡驱动，不要添加软链接！

* 路径可以使用自己目录 /home/xxx/cuda/ e.t.c.

#..一堆协议说明...
#直接按q退出协议说明.
accept/decline/quit: accept  #接受协议

You are attempting to install on an unsupported configuration. Do you wish to continue?
(y)es/(n)o [ default is no ]: yes #是否继续安装

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 396.37?
(y)es/(n)o/(q)uit: n  #是否显卡驱动包，由于已经安装显卡驱动，选择n
 
Install the CUDA 9.2 Toolkit?
(y)es/(n)o/(q)uit: y  #是否安装工具包，选择y
 
Enter Toolkit Location
 [ default is /usr/local/cuda-9.2 ]:  #工具包安装地址，默认回车即可
 
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: n #添加链接**注意这个连接，如果你之前安装过另一个版本的cuda，除非你确定想要用这个新版本的cuda，否则这里就建议选no，因为指定该链接后会将cuda指向这个新的版本**
 
Install the CUDA 9.2 Samples?
(y)es/(n)o/(q)uit: y  #安装样例
 
Enter CUDA Samples Location
 [ default is /home/ts ]: #样例安装地址默认即可
 
 ....
 
# ***安装完成***

3. CUDNN 解压并 'mv' 到CUDA对应目录

4. 更新bashrc路径

# Step 1: vim ~/.bashrc

#####ADD CUDA PATHS BEFORE CONDA ENV ACTIVATION########

# CUDA
export LD_LIBRARY_PATH="/home/xxx/cudas/cuda-9.2/lib64/"
export LIBRARY_PATH="/home/xxx/cudas/cuda-9.2/lib64/"
export CUDA_HOME="/home/xxx/cudas/cuda-9.2/"
export PATH="/home/xxx/cudas/cuda-9.2/bin:$PATH"

conda activate masktrack

#######################################################

# Step 3: source ~/.bashrc

5. 验证 用Python或者nvcc -V

import torch
import torch.utils
import torch.utils.cpp_extension
torch.utils.cpp_extension.CUDA_HOME  #输出 Pytorch 运行时使用的 cuda 

'''
>>> import torch.utils.cpp_extension
CUDA_HOME FINDING RESULTS: /home/xxx/cudas/cuda-9.2/
>>> torch.utils.cpp_extension.CUDA_HOME
'/home/xxx/cudas/cuda-9.2/'
'''