Replies: 1 comment
-
I solved it through https://www.youtube.com/watch?v=vBfM5l9VK5c&t=342s. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I want to train a gpt2 model and I have a GeForce RTX 2060 GPU , my os is windows
I installed cuda from the NVIDIA directory: https://developer.nvidia.com/cuda-downloads and use
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
I am really confused and I want to use my laptop to coding and every time it used CPU instead of GPU.
I tried many solution to activate the GPU but I still have the same problem. I need the right steps to fix it, please.
Beta Was this translation helpful? Give feedback.
All reactions