터미널 열고 아래 명령어 입력
python -m torch.utils.collect_env
아래와 같이 출력됨
Versions of relevant libraries:
[pip3] numpy==1.24.4
[pip3] torch==2.4.1
[pip3] torchvision==0.19.1
[pip3] triton==3.0.0
[conda] numpy 1.24.4 pypi_0 pypi
[conda] torch 2.4.1 pypi_0 pypi
[conda] torchvision 0.19.1 pypi_0 pypi
[conda] triton 3.0.0 pypi_0 pypi
'Code > Pytorch' 카테고리의 다른 글
| [Pytorch] Batch 개수, Data 개수 (0) | 2023.05.14 |
|---|---|
| [Pytorch] Seed 고정 (0) | 2023.05.09 |
| [Pytorch] GPU에 맞는 CUDA version 설치 (0) | 2023.04.14 |
| [Pytorch] Tensor 에서 특정 index 의 값을 뽑아 새로운 tensor 정의 (0) | 2023.04.12 |
| [Pytorch] .detach().cpu().numpy() (0) | 2023.04.10 |