登录有GPU的机器。 然后使用docker进入安装了cuda的环境。 查看gpu使用状况 查询命令介绍 启动代码中选择gpu进行运算 import os os.environ['CUDA_VISIBLE_DEVICES'] = '1'