TORCH_CUDA_ARCH_LIST

目录

查看方法：

设置方法：

---

查看方法：

python -c "import torch; print(torch.cuda.get_device_capability())"

设置方法：

1. 通过代码设置： 在你的 Python 脚本中，使用 os.environ 来设置 TORCH_CUDA_ARCH_LIST。例如，如果你想编译支持 CUDA 6.1 和 CUDA 7.5 的代码，可以在代码开始时加入以下内容：

   import os os.environ['TORCH_CUDA_ARCH_LIST'] = '6.1;7.5'

2. 通过命令行设置： 你也可以在命令行中设置这个环境变量，具体取决于你使用的操作系统。

   • Linux/macOS:

     bash

     export TORCH_CUDA_ARCH_LIST="6.1;7.5"

   • Windows:

     bash

     set TORCH_CUDA_ARCH_LIST=6.1;7.5