Linux下安裝pytorch的GPU版本

查詢cuda版本

cat /usr/local/cuda/version.txt

根據官網提示，用conda安裝：

conda create -n bartner python=3.7.4
conda activate bartner
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.0 -c pytorch