判斷 pytorch 是否使用 GPU

python -c "from __future__ import print_function; import torch; print(torch.cuda.is_available())"

發佈留言

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *