查询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
Linux下安装pytorch的GPU版本
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。