如何判断pytorch使用了gpu
答案:2 悬赏:40 手机版
解决时间 2021-02-08 21:35
- 提问者网友：最爱你的唇
- 2021-02-08 12:51
如何判断pytorch使用了gpu
最佳答案
- 五星知识达人网友：执傲
- 2021-02-08 13:25
可以试试以下代码
import torch
torch.__version__
torch.cuda.is_available()
我的环境是win10+anaconda+pytorch+cuda+cudnn
import torch
torch.__version__
torch.cuda.is_available()
我的环境是win10+anaconda+pytorch+cuda+cudnn
全部回答
- 1楼网友：我住北渡口
- 2021-02-08 14:21
pytorch动态计算图有什么好处 1.数据计算 Torch 自称为神经网络界的 Numpy, 因为他能将 torch 产生的 tensor 放在 GPU 中加速运算 (前提是你有合适的 GPU), 就像 Numpy 会把 array 放在 CPU 中加速运算。Torch和Numpy之间可以进行自由的切.
我要举报
如以上问答信息为低俗、色情、不良、暴力、侵权、涉及违法等信息，可以点下面链接进行举报！
大家都在看
推荐资讯