3090顯卡(CUDA11.1)安裝Pytorch


由於官方暫時沒有11.1版本的Pytorch安裝方法,可以使用11.0版本的安裝方法,經過測試,在3090上是可以正常使用的。
官方給出的安裝方法:
pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

在安裝過程中,由於連接的是Pytorch的官方地址,所以下載的速度是很慢的,可以手動下載torch包和torchvison的包進行安裝,運行一下上面的安裝命令,就能看到地址。
另外,安裝完torch包以后,在安裝torchvison的包時,可能會自動下載另一個版本的torch包,可以通過
pip install --no-deps torchvision-0.8.2+cu110-cp38-cp38-linux_x86_64.whl解決。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM