Mael-zys/T2M-GPT

Environment file

Closed this issue · 2 comments

tommyshelby4 commented

Could you please provide an environment.yml file compatible for CUDA 11.2+? My GPU does not support CUDA 10.x, causing me a series of problems. I would be truly grateful. Thanks in advance.

Jiro-zhang commented

Hi~
We have not tried cuda version 11.2, but have tried 11.1, and you can update the environment with the following command：

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
  • ❤️1
tommyshelby4 commented

Unfortunately, this conda update commmand does not seem to work.