ImportError

[gebaltso]

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.