hxiaoyuan 发表于 2023-10-12 21:31:04

Torch not compiled with CUDA enabled

求助,这是我的torch版本1.13.1+cpu,请问要怎么解决这个问题

isdkz 发表于 2023-10-12 21:31:19

这个问题的原因是你的Torch版本没有启用CUDA。要解决这个问题,你需要重新编译Torch并启用CUDA。

以下是解决问题的步骤:

1. 确保你已经安装了CUDA和cuDNN。你可以在NVIDIA的官方网站上下载这些软件。

2. 下载Torch的源代码并解压缩。

3. 打开终端并进入Torch的源代码目录。

4. 运行以下命令来编译Torch并启用CUDA:

      export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
   sudo python setup.py install
   

   这个命令会编译Torch并启用CUDA。请注意,这个过程可能需要一些时间。

5. 安装完成后,你可以运行以下命令来测试Torch是否已经启用了CUDA:

      th -e "require 'cutorch'; print(cutorch)"
   

   如果输出结果中包含了CUDA相关的信息,那么说明你已经成功启用了CUDA。

希望这个回答能够帮到你!

Mike_python小 发表于 2023-10-12 21:32:38

问题分析:
根据您的描述,您在使用torch库时遇到了"Torch not compiled with CUDA enabled"的错误。这个错误通常是由于您的torch版本没有启用CUDA支持导致的。

解决方案:
要解决这个问题,您可以尝试以下步骤:

1. 确认CUDA是否正确安装:
   - 首先,请确保已经安装了与您的torch版本相对应的CUDA工具包。您可以查看torch官方文档,确认您的torch版本和CUDA版本的兼容性。
   - 然后,确认CUDA的环境变量是否正确设置。可以检查PATH和LD_LIBRARY_PATH环境变量,确保它们包含了CUDA相关的路径。

2. 安装启用CUDA支持的torch版本:
   - 如果您的系统配置支持CUDA,并且您希望使用CUDA加速,请尝试安装启用CUDA支持的torch版本。
   - 您可以通过在命令行中运行以下命令来安装PyTorch的CUDA版本:
   
   pip install torch==1.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
   
   请注意,上述命令中的1.3.1+cpu是一个示例版本,您可以将其替换为您需要安装的特定版本。

3. 使用CPU运行代码:
   - 如果您不需要使用CUDA加速,可以修改代码以使用CPU运行。只需要将所有涉及CUDA的代码部分修改为CPU版本的实现即可。
   - 您可以通过检查代码中是否有使用torch.cuda相关函数来确定是否涉及CUDA操作。

希望以上解决方案能够帮助您解决问题。如果您还有其他问题,请随时提问。

球一个最佳答案谢谢啦!这对我非常重要!{:10_254:}{:10_254:}{:10_254:}

hxiaoyuan 发表于 2023-10-15 15:09:46

isdkz 发表于 2023-10-12 21:31
这个问题的原因是你的Torch版本没有启用CUDA。要解决这个问题,你需要重新编译Torch并启用CUDA。

以下是 ...

我的是windows,有没有windows的代码呀

isdkz 发表于 2023-10-15 15:14:01

hxiaoyuan 发表于 2023-10-15 15:09
我的是windows,有没有windows的代码呀

你的是N卡不?

hxiaoyuan 发表于 2023-10-15 15:43:37

isdkz 发表于 2023-10-15 15:14
你的是N卡不?


是英伟达的卡

hxiaoyuan 发表于 2023-10-15 15:44:56

Mike_python小 发表于 2023-10-12 21:32
问题分析:
根据您的描述,您在使用torch库时遇到了"Torch not compiled with CUDA enabled"的错误。这个错 ...

怎么把有cuda的代码修改为CPU版本的呀

Mike_python小 发表于 2023-10-15 15:45:37

hxiaoyuan 发表于 2023-10-15 12:44
怎么把有cuda的代码修改为CPU版本的呀

什么意思?

hxiaoyuan 发表于 2023-10-15 16:00:19

Mike_python小 发表于 2023-10-15 15:45
什么意思?

您说“如果您不需要使用CUDA加速,可以修改代码以使用CPU运行。只需要将所有涉及CUDA的代码部分修改为CPU版本的实现即可。”怎么修改呢
页: [1]
查看完整版本: Torch not compiled with CUDA enabled