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
