pytorch指定使用GPU编号
方法1：import osos.environ["CUDA_VISIBLE_DEVICES"] = "1"#使用1号方法2：手动敲入CUDA_VISIBLE_DEVICES = 1
·
方法1：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1" #使用1号
方法2：手动敲入
CUDA_VISIBLE_DEVICES=0 python your_script.py
汇聚全球AI编程工具，助力开发者即刻编程。
更多推荐
1
0- 0
已为社区贡献1条内容
所有评论(0)