最近開始學習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