CUDA_HOME environment variable is not set. Please set it to your CUDA install root

yang-song / score_sde_pytorch

PyTorch implementation for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

https://arxiv.org/abs/2011.13456

Apache License 2.0

1.58k stars 295 forks source link

Open forever208 opened 10 months ago

forever208 commented 10 months ago

error when using main.py script to do evaluation

f-krause commented 8 months ago

I have the same issue, would appreciate some help (i.p. a list of requirement versions compatible for python3.9)

GuJiaWen777 commented 6 months ago

maybe export CUDA_HOME=$CONDA_PREFIX