pytorch使用过程中的cmd

2022-10-23 10:36:23

pytorch使用过程中的cmd

1.nvidia-smi
证明驱动已装好，并显示GPU的内存、使用情况等信
息。
在这里插入图片描述
2.torch.cuda.is_available() #判断当前GPU是否可用
在这里插入图片描述

  • 作者：Giao哥不瘦到100不改名
  • 原文链接：https://blog.csdn.net/exsolar_521/article/details/120271471
    更新时间：2022-10-23 10:36:23