1 PyTorch

2 GPU

2.1 find CUDA version on Windows

on Windows CMD

nvcc --version
or
nvidia-smi

or

right click desktop -> Navida control panel -> Help -> System information -> component -> Navida CUDA driver version

or

Anaconda Prompt -> type in “numba -s” -> check __CUDA Information__

2.2 install GPU version of Pytorch

go to Pytorch website -> select the pytorch version with the CUDA version -> generate installation command e.g. conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

3 Hugging Face

Where does hugging face’s transformers save models? C:\Users\username\.cache\huggingface\transformers

Home Page