在conda环境中安装nvcc/cuda-toolkit

前言

作为赛博洁癖，新配的服务器不想管理复杂的cuda环境，因此正在测试除了驱动外其他所有的cuda相关组件通过conda管理的可行性，因此产生此文作为配置记录方便后续查询。本文不针对特定版本做管理，默认以最新版安装。

Conda可以安装什么？

除了ffmpeg可以直接用conda安装之外，nvcc等一系列工具都可以直接通过conda管理，实现多个环境之间隔离。以下内容均在激活的conda环境内进行，表示在该环境下安装工具。

nvcc

cuda 编译工具

conda install -c nvidia cuda-nvcc

安装之后通过nvcc -V检查是否安装成功

• 在安装DeepSpeed时出现如下错误，发现是没有nvcc，通过在conda内直接安装nvcc解决

  CUDA_HOME does not exist, unable to compile CUDA op(s)

cuda-compiler

cuda 相关的包，具体安装的包如图：

conda install -c nvidia cuda-compiler

cuda-toolkit

⚠️注意不是cudatoolkit无减号版，这是两个不同的包

conda install -c nvidia cuda-toolkit