在终端运行python指定显卡
比如我要跑的程序是train.py，然后我指定4号显卡去跑就↓
CUDA_VISIBLE_DEVICES=4 python train.py
效果等同于在代码里加
import os
os.environ['CUDA_VISIBLE_DEVICES']='4'
配置网络代理
os.environ['http_proxy']='127.0.0.1:7890'
os.environ['https_proxy']='127.0.0.1:7890'
代码环境配置
于 2023-03-06 21:31:09 首次发布
通过设置环境变量CUDA_VISIBLE_DEVICES为4，可以指定使用4号GPU运行Python程序train.py，这在深度学习中常见，确保计算任务在特定显卡上执行。等效的代码实现是在Python脚本中导入os模块并设置环境变量。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
