Linux에 Pytorch 설치하기.
2021. 11. 9. 11:50ㆍ프로그래밍을 위한 기본 세팅/Ubuntu 서버 세팅
Pytorch 홈페이지(https://pytorch.org/)에 가면 원하는 사양에 맞춰서 설치 할 수 있다.
Previous versions of PyTorch를 누르면 이전 버전 Pytorch 들을 확인 할 수 있다.
Linux Conda Python CUDA 11.3
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
Linux Conda Python CUDA 10.2
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
'프로그래밍을 위한 기본 세팅 > Ubuntu 서버 세팅' 카테고리의 다른 글
| [Linux] PyCharm 설치 (0) | 2021.11.19 |
|---|---|
| linux 그래픽 드라이버(Nvidia driver) 설치 (0) | 2021.11.15 |
| RTX 30xx 시리즈 tensorflow 1.15 (구버전) 설치 방법 (0) | 2021.11.08 |
| 싱글, 병렬 GPU stress test (0) | 2021.11.05 |
| Nvidia-smi 실시간으로 확인하기 (0) | 2021.11.05 |