분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
pytorch 설치(CUDA버전 포함)
Pytorch 처음 설치할 때 매번 검색하기에 번거로울 수 있으니 공유합니다.
1️⃣ PyTorch (CPU 버전 설치)
pip install torch torchvision torchaudio
2️⃣ PyTorch (CUDA 버전 설치)
※ CUDA 버전은 사용하는 GPU 환경에 따라 맞춰 설치해야 합니다.
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
버전 모르면 nvidia-smi로 확인한번 해보세요~
3️⃣ 설치 확인
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"
pytorch 한번 꼬이면 지웠다가 다시까시는 걸 추천합니다.
그게 제일 빠르더라구요 ㅎ
DACON Co.,Ltd | CEO Kookjin Kim | 699-81-01021
Mail-order-sales Registration Number: 2021-서울영등포-1704
Business Providing Employment Information Number: J1204020250004
#901, Eunhaeng-ro 3, Yeongdeungpo-gu, Seoul 07237
E-mail dacon@dacon.io |
Tel. 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved