프로그래밍을 위한 기본 세팅/Ubuntu 서버 세팅
Linux에 Pytorch 설치하기.
Mason-CHOI
2021. 11. 9. 11:50
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