使用命令首先查看服务器显卡的使用情况
os.environ[“CUDA_VISIBLE_DEVICES”] = “0”
使用此命令可以第几块显卡进行加速,直接放置于程序最上面即可。
深度学习使用GPU加速
最新推荐文章于 2024-06-05 00:40:24 发布
使用命令首先查看服务器显卡的使用情况
os.environ[“CUDA_VISIBLE_DEVICES”] = “0”
使用此命令可以第几块显卡进行加速,直接放置于程序最上面即可。