CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=2,3
  1. 仅仅支持当前显卡在环境中可见。
  2. 在命令行的使用:加在运行代码之前即可。
  3. 使用该命令后代码中的cuda id即失效,因为在当前环境下,只有你指定的显卡可见了,其他的id已不可见。
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐