ubuntu中报错AssertionError: Torch not compiled with CUDA enabled

在Ubuntu中，当你在使用Torch库时遇到"AssertionError: Torch not compiled with CUDA enabled"错误时，这通常是因为你的Torch库没有启用CUDA支持。

CUDA是NVIDIA提供的用于并行计算的平台和编程模型，它可以利用GPU的强大计算能力来加速深度学习任务。要解决这个问题，你需要重新编译Torch库并启用CUDA支持。

以下是一些可能的步骤：

1. 确保你已经正确安装了NVIDIA显卡驱动和CUDA工具包。你可以通过运行以下命令来检查CUDA是否正确安装：

   nvcc --version

如果显示了CUDA的版本信息，则表示CUDA已经正确安装。

2. 确保你已经正确安装了Torch库。你可以通过运行以下命令来检查Torch是否已经安装：

   import torch

如果没有报错，则表示Torch已经正确安装。

3. 如果你已经安装了正确版本的CUDA和Torch，但仍然遇到该错误，那么可能是因为你的Torch库没有启用CUDA支持。你可以尝试重新编译Torch库并启用CUDA支持。具体步骤如下：

- 首先，确保你已经安装了CMake和GCC编译器。你可以使用以下命令来安装它们：

     sudo apt-get install cmake gcc

- 然后，克隆Torch库的源代码并进入源代码目录：

     git clone https://github.com/pytorch/pytorch.git
     cd pytorch

- 创建一个用于构建的新目录，并进入该目录：

     mkdir build
     cd build

- 运行CMake命令来配置构建过程，并启用CUDA支持：

     cmake .. -DCMAKE_PREFIX_PATH=/path/to/cuda -DCMAKE_C_COMPILER=gcc -DCMAKE_CXX_COMPILER=g++

其中，`/path/to/cuda`是你CUDA安装的路径。

- 运行make命令来编译Torch库：

     make

- 最后，运行make install命令来安装Torch库：

     sudo make install

完成上述步骤后，重新运行你的代码，应该就不会再出现"AssertionError: Torch not compiled with CUDA enabled"错误了。

希望以上解答对你有帮助！如果你还有其他问题，请继续提问。