自己电脑cuda driver版本是11.1的情况下 如何装pytorch的gpu版本?
自己电脑cuda driver版本是11.1的情况下 如何装pytorch的gpu版本(因为pytorch只支持到10.2的cuda版本)
按照官网的命令安装就可以了,之前没有11.0的时候可以按照源码编译
pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html