直接使用 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。
直接使用 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。