$ import torch
$ torch.cuda.is_available() ( cuda 먼저 확인 / False 뜨면 gpu 버전에 맞게 pytorch 재설치 )
$ torch.cuda.get_device_name()
'Setting, Error > Python, Python Library' 카테고리의 다른 글
| Tensorflow, pyTorch - 원하는 GPU 할당 (0) | 2023.06.08 |
|---|---|
| Jupyter Notebook 자동완성 Tab 안 될 때 (0) | 2022.02.13 |
| RTX 3090 Tensorflow GPU 동작확인 (0) | 2021.11.23 |
| [Install] RTX 3090, cuda 11.2 Tensorflow-gpu 설치 (0) | 2021.11.23 |
| [Install] RTX 3090, cuda 11.2 Pytorch 설치 (0) | 2021.08.09 |