cuda pytorch torchvision版本對應問題


最近開始學習nlp,安裝fairseq,要求pytorch版本>=1.5,而服務器的cuda版本為10.0和9.2的,直接通過conda安裝會自動安裝對應cuda10.2的pytorch,影響使用。現記錄解決這一問題的過程。

1. cuda,pytorch,torchvision的對應關系如下圖:

2. pytorch torchvision下載庫

https://download.pytorch.org/whl/cu92/torch_stable.html

3. 安裝cuda9.2對應的pytorch1.7.1

wget https://download.pytorch.org/whl/cu92/torch-1.7.1%2Bcu92-cp37-cp37m-linux_x86_64.whl
wget https://download.pytorch.org/whl/cu92/torchvision-0.8.2%2Bcu92-cp37-cp37m-linux_x86_64.whl
pip install torch-1.7.1+cu92-cp37-cp37m-linux_x86_64.whl torchvision-0.8.2+cu92-cp37-cp37m-linux_x86_64.whl


免責聲明!

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



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