conda环境下使用nvcc -V报错nvcc: command not found的一种解决方法
前言
缘起
实验室的学弟问我为什么他使用nvcc命令报错,起先我以为他用的是老师给的root账户,按照参考文献1便可以解决问题。
但由于并非root用户,/usr/local下没有cuda,于是便无法按照参考1中的方法去做。
这里提供一种方法,其实是参考了文献2,但似乎是歪打正着,因为2要解决的问题和我的不一样O_o。
问题
使用nvcc -V报错如下
解决方法
报错原因在于当前conda环境没有安装nvcc,于是使用如下命令安装即可
conda install -c nvidia cuda-nvcc
补充:执行上述命令后会默认安装cuda-nvcc的最新版本,于是这里涉及到cuda-nvcc版本号的确定问题,需要根据实际情况确定,下面我举一个例子。
我在安装apex时有遇到如下报错
此时需要重新安装cuda-nvcc,版本号需要与安装pytorch的cuda版本号一致,以上图为例,需要安装cuda-nvcc11.3的版本,执行如下命令即可,不需要卸掉之前的cuda-nvcc,安装时会自动进行cuda-nvcc版本迁移。
conda install -c "nvidia/label/cuda-11.3.0" cuda-nvcc
安装其他版本的cuda-nvcc命令参照官网。
再次使用nvcc -V,成功啦!^ _ ^
补充:上面我用的我的pruner环境,安装了nvcc,因为不想动base环境~此外,上图中的cuda-nvcc版本号是12.1,这是我直接执行conda install -c nvidia cuda-nvcc
的结果。
后面我重新安装了cuda-nvcc11.3版本,执行conda install -c "nvidia/label/cuda-11.3.0" cuda-nvcc
cuda-nvcc的版本号为11.3,结果如下所示
参考文献
分类:
Programming
标签:
Python
, Remote Server
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· winform 绘制太阳,地球,月球 运作规律
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人