3090 cuda环境配置杂记

实验室新配了3090的电脑，cuda环境配置上出了些问题，百度很久才找到解决方法，因此记录下来。

主要参考：(6条消息) 服务器Linux环境配置cuda（非管理员）_@@Lynn的博客-CSDN博客_服务器配置cuda；(6条消息) nvcc fatal : Unsupported gpu architecture ‘compute_86‘问题解决_chen_yanan的博客-CSDN博客

服务器Linux环境配置cuda路径

报错：OSError: CUDA_HOME environment variable is not set. Please set it to your CU...

解决方案：

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

gedit ~/.bashrc # 打开bashrc

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

# 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

nvcc fatal : Unsupported gpu architecture ‘compute_86‘

显卡与cuda版本支持的算力不匹配。

解决方案：限制算力,追加一行：

打开

gedit ~/.bashrc

追加