pytorch cuda run `false`/ pytorch cant run

Condition

RUN pyTorch時一直跑到cpu而非gpu，造成Training異常久。

Solution

Step1. 輸入nvidia-smi，看是否有裝cuda及nvidia driver

Step2. 輸入nvcc -V，看是否有裝NVIDIA CUDA Complier，通常裝完cuda就會有這個了

Step3. 在Terminal 跑pytorch，確認當前是跑cpu還是gpu，>>>True 才是跑GPU

import torch ## import package

print(torch.cuda.is_available()) #確認有無使用cuda

exit() #跳出python terimal

Step4. 確認conda安裝的package是否正確

我當初是直接conda install pytorch==(version)，後來才知道要到官網找對應的指令，不然會裝到cpu版本的pytorch
官網: Pytorch
取得對應的conda command，這樣應該就能順利運行cuda in python