解决Anaconda虚拟环境中Conda安装cudatoolkit之后,which nvcc还是系统级cudatoolkit问题。
conda环境中虽然安装了cudatoolkit ,但缺少nvcc编译器。这是因为标准cudatoolkit包不包含编译工具,需要安装cudatoolkit-dev包。为什么会出现这种情况呢?
·
conda环境中虽然安装了cudatoolkit ,但缺少nvcc编译器。这是因为标准cudatoolkit包不包含编译工具,需要安装cudatoolkit-dev包。为什么会出现这种情况呢?
Conda与NVIDIA官方完整CUDA Toolkit之间存在本质区别
功能定位不同
- 标准cudatoolkit:仅包含运行预编译CUDA程序所需的最小动态链接库
- 完整cudatoolkit:包含编译器(nvcc)、调试器、头文件等完整开发工具链
设计目的差异
- conda版cudatoolkit主要服务于PyTorch/TensorFlow等框架的运行时需求
6
- 开发版cudatoolkit-dev才包含编译自定义CUDA扩展所需的工具
3
安装体积考量
- 标准cudatoolkit约500MB,而完整CUDA Toolkit超过3GB,conda默认选择轻量级方案
- 含编译器(nvcc)、调试器、头文件等完整开发工具链
解决方案
在自己的虚拟环境中运行以下代码
conda install -c conda-forge cudatoolkit-dev=11.7 //安装开发版工具链,根据自己的需求换
//设置临时环境变量
export CUDA_HOME=$CONDA_PREFIX
export PATH=$CUDA_HOME/bin:$PATH
//验证版本
which nvcc # 应显示conda环境路径
nvcc --version # 应输出11.7版本

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)