下面命令是查看是否使用GPU

python #命令行输入
import torch  #导入torch包
torch.cuda.is_available()  #查看是否使用GPU，返回True证明使用，Falsea不能使用

# pytorch # 深度学习
Logo
2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

cover

毕设神器，当Java初学者遇到飞算，仿佛直接打通了任督二脉

avatar 2048 AI社区
cover

科研论文必备：九大查重系统功能差异与操作技巧

avatar 2048 AI社区
cover

高效学术写作：九大查重平台深度对比与应用策略

avatar 2048 AI社区