微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何让cuda回家?我对 conda 虚拟环境感到困惑

如何解决如何让cuda回家?我对 conda 虚拟环境感到困惑

我试图在 pytorch 中训练 DDP。但是nccl插件错误。 所以我尝试运行 nccl-test 以确保我的服务器正常工作。

所以我需要 CUDA_HOME、Nccl_HOME。但是,我无法确定哪个是 CUDA_HOME,因为 sudo find / -name cuda 输出了许多包含 cuda 的 conda 环境路径。

并且 /usr/local/ 中不存在 cuda,但是 /usr/lib/cuda 确实存在并且 nccl.h 无处可寻...

重新安装 cuda 是最好的方法吗?

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。