NVlabs / stylegan3

Official PyTorch implementation of StyleGAN3
Other
6.28k stars 1.1k forks source link

ImportError #239

Open gebaltso opened 1 year ago

gebaltso commented 1 year ago

Hello! When I run sh scripts/gui.sh I get the following error: Traceback (most recent call last): File "/home/.../DragGAN/visualizer_drag.py", line 14, in import torch File "/home/.../anaconda3/envs/stylegan3/lib/python3.9/site-packages/torch/init.py", line 229, in from torch._C import * # noqa: F403 ImportError: /home/.../anaconda3/envs/stylegan3/lib/python3.9/site-packages/torch/lib/libc10_cuda.so: undefined symbol: cudaMemPoolSetAttribute, version libcudart.so.11.0

Is it perhaps a problem with pytorch version or something? Thanks in advance.