【Python】set os.environ[“CUDA_VISIBLE_DEVICES“] = ‘1‘ Invalid

RockWang.

已于 2024-08-27 18:03:28 修改

阅读量212

点赞数 1

分类专栏：程序语言文章标签： python 开发语言

于 2024-08-27 18:02:42 首次发布

本文为博主原创文章，未经博主允许不得转载。

本文链接：https://blog.csdn.net/xi_shui/article/details/141609310

版权

程序语言专栏收录该内容

53 篇文章 6 订阅

订阅专栏

If set os.environ[“CUDA_VISIBLE_DEVICES”] = ‘1’ Invalid

you can place the code block os.environ["CUDA_VISIBLE_DEVICES"] = '1' before all cuda code calls.

For example:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '2'
import numpy as np
import torch
import cv2
from diffusers import (AutoencoderKL, DDIMScheduler,
                       DPMSolverMultistepScheduler,
                       EulerAncestralDiscreteScheduler, EulerDiscreteScheduler,
                       PNDMScheduler)
from omegaconf import OmegaConf