本文介绍三种设置CUDA_VISIBLE_DEVICES的方法：1. 在代码中通过os.environ指定；2. 在命令行直接设置；3. 通过export命令配置。这些方法用于控制CUDA使用的GPU设备。
1、在代码里面实现
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '2,3'
2、在命令行中指定
CUDA_VISIBLE_DEVICES=2 python xxx.py
3、在某一个命令窗口指定
export CUDA_VISIBLE_DEVICES="xx"
tensorflow训练时指定GPU
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
