hans

hans

【其他】使用Pip安裝Pytorch


直接使用 pip install pytorch 只能安裝 CPU 版本,無法使用 GPU,請使用以下指令進行安裝。

pip3 install torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

1.4.0 是版本號,如果輸入錯誤也沒關係,系統會告訴您可用的版本。

cu100 表示 CUDA 版本 10.0。

載入中......
此文章數據所有權由區塊鏈加密技術和智能合約保障僅歸創作者所有。