PyTorch安裝慢,加速PyTorch的方法

不着急的安裝方法:

conda install pytorch torchvision torchaudio -c pytorch
or
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

趕時間的安裝方法:

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章