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