- 博客(4)
- 收藏
- 关注
RSS订阅原创 在macos中的pytorch如何使用gpu训练模型
最近在学pytorch的过程中，看到不少的代码中含有data.cuda()等，其实就相当于将data放到GPU用于后续的计算。于是我在百度了macos如何使用cuda，最后没有解决，并且在英伟达官网也没有看到cuda支持macos，也许是cuda不支持macos？那么macos到底有没有gpu呢，应该是有的，见下图。 但是实在是没有找到使用gpu的解决办法，后来，我在网上看到了可以使用cpu代替gpu进行训练。代码如下。这里将device设置为cpu，在后续使用.cuda(
2024-09-19 20:38:19
657
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人