包括共享显存和专用显存
对于不支持DX12的显卡,查看方法,GPU显存只能用专用显存
对于支持DX12的显卡,安装tensorflow-cpu/pytorch-cpu和微软的tensorflow-directml-plugin/pytorch-directml-plugin,就会用DirectX 12运行,深度学习框架就可以像游戏一样使用共享显存。
查看显存方法
- 任务管理器
- win+R输入dxdiag的显示可看
参考
https://zhuanlan.zhihu.com/p/408897489