安裝好cuda 和pytorc之後,如何判斷 GPU是可用的??

c) 驗證GPU是否可用

python
import torch 
print(torch.cuda.is_available())

返回True,表明已經可以使用GPU了.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章