复现代码,要求cuda=11.8,但当前系统cuda=11.1。在NVIDIA 驱动支持的前提下,进行以下步骤:
1. 创建一个新环境,例如:conda create -n xxx python=3.8
2. 在 conda 环境中安装特定版本的 CUDA 工具包
conda install --channel "nvidia/label/cuda-11.8.0" cuda
3. 查看conda环境内的nvcc
ls $CONDA_PREFIX/bin/nvcc 2>/dev/null || echo "No nvcc in conda env"
which -a nvcc # 查看所有可用的 nvcc
4. 此时nvcc --version会出现cuda版本为11.1的情况。我们要让 nvcc 使用 conda 环境内的版本,即:
export PATH="$CONDA_PREFIX/bin:$PATH" # 临时设置(当前终端有效)
5. 此时,
python -c "import torch; print(torch.__version__)"
nvcc --version
均为11.8
1159

被折叠的 条评论
为什么被折叠?



