自己電腦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