우분투 20.04 pythorch 설치

nvidia-smi 명령어 입력시 표가 뜬다면 설치된 것

bash Anaconda3-2020.02-Linux-x86_64.sh
아나콘다 사이트가서 linux 64bit 버전 설치

conda create -n pytorch python=3.7
conda activate pytorch
환경 활성화시 terminal 앞에 pytorch라는 것이 나타남.

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
파이토치 가상환경이 activate되어 있는 상태에서 파이토치 설치 진행

파이토치 가상환경 activate 상태에서
python
import torch
torch.__version__
torch.version.cuda
exit()
순서대로 진행하면 확인할 수 있다.