nshepperd/gpt-2

GPT2 Fine Tuning does not support Ampere (RTX 3000s) Cards

imakemoneymoves opened this issue · 0 comments

Hello, I have installed in my system a 3080 OC Edition. However, this program always choose to use the CPU instead of GPU. GPU is detected when i ran a nvidia-smi check and I set CUDA_VISIBLE_DEVICES=0. It just completely ignores the card. Any suggestions?