GPU - Githubissues

chanind / frame-semantic-transformer

Frame Semantic Parser based on T5 and FrameNet

https://chanind.github.io/frame-semantic-transformer

MIT License

51 stars 10 forks source link

GPU #13

Open sdspieg opened 1 year ago

sdspieg commented 1 year ago

Any suggestions for which versions of Tensorflow, CUDA and cuDNN to run?

chanind commented 1 year ago

This library uses PyTorch under the hood rather than Tensorflow, but other than that whatever version of CUDA will work with the version of PyTorch you want to install should be fine. It looks like the most recent version of PyTorch (1.13.1) works with CUDA 11.6 or 11.7 (https://pytorch.org/get-started/locally/#start-locally).

sdspieg commented 1 year ago

Just sharing progress (from a Windows machine with a GPU)

conda create --name pytorch python=3.10
conda activate pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

That SEEMS to work - I see that my gpu is using 6GB of VRAM. Not sure about time improvement, though...