linux中指定gpu Id
reference:linux服务器如何指定gpu以及用量方法1:在命令行中CUDA_VISIBLE_DEVICES=0方法2:在python文件中import osos.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定第一块gpu...
·
reference:linux服务器如何指定gpu以及用量
方法1:在命令行中
CUDA_VISIBLE_DEVICES=0
方法2:在python文件中
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定第一块gpu
更多推荐
已为社区贡献2条内容
所有评论(0)