search

prajwalsingh / EEGStyleGAN-ADA

Pytorch code of paper "Learning Robust Deep Visual Representations from EEG Brain Recordings". [WACV 2024]
MIT License
27 stars 4 forks source link

Question: about pytorch version #26

Open gyuseok0917 opened 1 week ago

gyuseok0917 commented 1 week ago

Hello. I have a question about Pytorch version compatibility in relation to the use of the code. The markdown manual states that Pytorch 1.7.0 with CUDA 11.0 version is used when installing the library. My server environment uses Pytorch 1.13.0 with CUDA 11.6, and I would like to know if there is no problem in running the code without changing the version.

prajwalsingh commented 1 week ago

Hi @gyuseok0917, if you have the latest cuda install, there might be no issue. At that time, it had an older version, which is why there were some issues. The specific reason was that when I tried running the metric computation code, it was not working at that time.