ziplab / PTQD

The official implementation of PTQD: Accurate Post-Training Quantization for Diffusion Models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RuntimeError: No CUDA GPUs are available

mengmeexix opened this issue · comments

It is strongly recommended to remove the setting of cuda device ID in the code:
import os os.environ['CUDA_VISIBLE_DEVICES'] = '7'

as the error message No CUDA GPUs are available keeps running

Thx for your advice!