第一步、安裝 PyTorch 機器學習框架。
安裝CPU版本:
pip install torch==1.8.1+cpu torchvision==0.9.1+cpu torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
安裝GPU版本:
pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html第二步、輸入 pip list 檢查是否安裝成功。
第三步、執行計算時輸入 nvidia-smi 查看 Type 類型。
C : Compute = CUDA or OpenCL
G : Graphics = DirectX or OpenGL