설치 사이트
https://pytorch.org/get-started/locally/
cuda 버전 확인
파이참 터미널
nvidia-smi |
사이트에 원하는 항목 선택
conda 나 pip 원하는 항목 선택
가장 아래 커맨드 복사하여 파이참 터미널에 입력
설치시작
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 |
CPU 버전
pip3 install torch torchvision torchaudio |
설치 완료 확인
import torch # 텐서 생성 테스트 x = torch.rand(5, 3) print(x) # GPU 사용 가능 여부 확인 print(f"CUDA 사용 가능 여부: {torch.cuda.is_available()}") |
'공부 > 파이썬' 카테고리의 다른 글
[파이썬] qt designer 연결 (0) | 2022.10.06 |
---|---|
[파이썬] 프로젝트 생성 (0) | 2022.10.06 |
[파이썬] 설치 (0) | 2022.10.06 |