zl程序教程

您现在的位置是:首页 >  后端

当前栏目

python中使用指定GPU

Python 指定 GPU 使用
2023-09-27 14:23:28 时间

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "2"  # 或 os.environ["CUDA_VISIBLE_DEVICES"] = "0,2,3" 

(等号右边的双引号可以省略)