0
点赞
收藏
分享

微信扫一扫

查找电脑是否支持GPU&GPU使用

 

在任务管理器和设备管理器查看自己的gpu配置 

在代码中设置使用gpu运行代码

import os
# 使用第一张与第三张GPU卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2"

具体安装和配置可参考:

怎样知道自己的电脑是否支持GPU?_进阶女学霸-CSDN博客_怎么看电脑有没有gpu

CUDA指定GPU的使用方法_haima1998的专栏-CSDN博客_指定cuda

举报

相关推荐

0 条评论