服务器Linux环境配置cuda（非管理员）

服务器Linux环境配置cuda（非管理员）

首先查看一下/usr/local路径下cuda的版本，以cuda-11.1为例,

gedit ~/.bashrc # 打开bashrc

# 在bashrc末尾的位置添加以下代码
export PATH="/usr/local/cuda-11.1/bin/:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH"
export CUDA_HOME=/usr/local/cuda-11.1

# ctrl+s保存，关闭bashrc，在终端中输入以下代码，使修改后的bashrc生效
source ~/.bashrc

检查一下cuda是否可用：

conda activate my_env_name # 进入虚拟环境
python # 进入python
>>> import torch
>>> import torch.utils.cpp_extension
>>> print(torch.cuda.is_available())
True
>>> print(torch.utils.cpp_extension.CUDA_HOME)
/usr/local/cuda-11.1
>>> print(torch.version.cuda)
11.1
>>> from torch.backends import cudnn
>>> print(cudnn.is_available())
True