如何在conda环境中设置cuda路径？
如何在conda环境中设置cuda路径？
提问于 2022-06-20 08:19:47
我使用的是ubuntu 22 LTS。我想在conda虚拟环境中使用不同版本的Cuda。我想为tensorflow 1.x使用一个虚拟conda环境，为tensorflow 2.x使用另一个conda，为PyTorch使用一个环境。我在创建环境时添加了Cuda的版本。我的问题是，与其使用安装在conda (即。( conda create -n tf1 cudnn=7.6.5 cudatoolkit=10.1.243)，它使用的是系统cuda。当我键入
nvcc --version产出如下：
nvcc: NVIDIA (R) Cuda编译器驱动程序版权(c) 2005-2022 NVIDIA公司基于Tue_May__3_18:49:52_PDT_2022 Cuda编译工具，版本11.7，V11.7.64构建cuda_11.7.r11.7/编译器.31294372_0
如何设置使用安装在环境中而不是系统中的特定Cuda版本的路径？
回答 1
Stack Overflow用户
回答已采纳
发布于 2022-06-20 13:46:52
可以使用此命令设置CUDA_HOME环境变量。
conda env config vars set CUDA_HOME=""您应该能够使用此命令找到您的conda安装的CUDA库的路径。
sudo find / -name nvcc您还可能需要查看这些文档，以便将环境变量设置为激活步骤的一部分。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/72684130
复制相关文章
相似问题
腾讯云开发者
