PyTorch 使用笔记
因为最近在使PyTorch,在这记录一下PyTorch使用的技巧.
CUDA
在使用显卡时候需要用CUDA,model、和输入要CUDA。
注意:
- 直接TensorName.cuda()是不行的要 TensorName = TensorName.cuda()
因为最近在使PyTorch,在这记录一下PyTorch使用的技巧.
在使用显卡时候需要用CUDA,model、和输入要CUDA。
注意:
568
1723

被折叠的 条评论
为什么被折叠?