如何在win11种设置cuda_visible_devices
时间: 2023-12-13 09:03:53 浏览: 535
failed call to cuInit: CUDA_ERROR_NO_DEVICE解决方法
3星 · 编辑精心推荐
在 Windows 11 中设置 `CUDA_VISIBLE_DEVICES` 变量,可以按照以下步骤进行操作:
1. 首先,打开“开始菜单”,并搜索并选择“系统”。
2. 在系统设置页面中,选择“高级系统设置”。
3. 在弹出的窗口中,点击“环境变量”按钮。
4. 在环境变量窗口中,可以看到两个部分:用户变量和系统变量。如果您希望对当前用户生效,可以在用户变量部分进行设置;如果要对系统中的所有用户生效,则在系统变量部分进行设置。
5. 在对应的部分中,找到名为 `CUDA_VISIBLE_DEVICES` 的变量(如果不存在,则需要创建该变量)。
6. 点击“新建”按钮,在“变量名”中输入 `CUDA_VISIBLE_DEVICES`,在“变量值”中输入您希望设置的设备索引号。多个设备索引号之间使用英文逗号分隔。例如,如果您希望仅使用第一个和第三个设备,可以输入 `0,2`。
7. 点击“确定”按钮保存设置。
8. 重新启动计算机以使更改生效。
完成上述步骤后,`CUDA_VISIBLE_DEVICES` 变量将被设置为您指定的设备索引号,CUDA 应用程序将只会在指定的设备上运行。请确保您已正确安装 CUDA 驱动程序和相关的 GPU 计算库。
注意:在设置 `CUDA_VISIBLE_DEVICES` 变量时,请确保没有设置其他与 CUDA 相关的环境变量,以避免冲突。另外,一些应用程序可能具有自己的设备选择设置,这些设置可能会覆盖 `CUDA_VISIBLE_DEVICES` 变量的值。
阅读全文