解决CUDA10.1无法安装Pytorch GPU版的问题

今天在安装Ubuntu下pytorch环境时, 偶然发现现在Pytorch只支持CUDA10.2以上的版本, 而我的旧版本的CUDA10.1使用conda命令安装的话, 只能自动安装CPU版本, 找了好久才找到解决的方法, 就是输入如下命令

pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

亲测有效

你可能感兴趣的:(经验分享,ubuntu)