아무 생각 없이 import torch 를 수행하면서 pip install torch를 실행시켰는데,
이를 수행하니 파이토치가 GPU를 사용할 수 없었다.
추가적인 정보를 확인해보니, 전용 pytorch를 설치해줘야 GPU 사용이 가능했다.
먼저 해당 링크에 접속해준다.
Get Started
Set up PyTorch easily with local installation or supported cloud platforms.
pytorch.org
해당 링크에서 자신의 컴퓨터와 cuda 설치 정보와 맞는 pytorch를 찾아, 해당 Run this Command를 쉘에서 실행시켜준다.
cuda의 경우 최신 버전이 출시되더라도, pytorch는 그 버전을 stable로 지원하기까지 좀 시간이 걸린다.
물론 Nightly로 지원은 빠르게 해주긴 하지만, 이 글을 검색해서 찾아왔다면 아마 Nightly 버전을 사용하긴 어려울 테니?
꼭 PyTorch가 지원하는 버전으로 cuda를 다운받아 사용해주자.
참고로 자신의 cuda 버전은 다음 명령어를 통해 확인 가능하다.
nvidia-smi
그러면 콘솔로 print문을 찍어봤을 때, 다음과 같은 문구를 볼 수 있게 된다.
'Article - 깊게 탐구하기 > OpenAI Gym' 카테고리의 다른 글
| MountainCar - 단순 DQN으로 풀기 (0) | 2025.09.06 |
|---|