
pytorch
leogo17
这个作者很懒,什么都没留下…
展开
-
pytorch: grad is None while training
原因可能是网络定义过程中存在没有进行前向计算的层,当计算loss时这些层没有参与运算,因此梯度无法传播到这些层解决办法:for p in model.parameters(): if p.grad is not None: print(p.grad.data)...原创 2018-04-17 10:24:10 · 12313 阅读 · 3 评论 -
pytorch使用中遇到的问题
1. load模型参数文件时,提示torch.cuda.is_available() is False。按照pytorch官方网页又安装了一次pytorch,而不是直接使用清华源,执行pip install torch,暂时不知道为什么。2. 使用CrossEntropyLoss时,要求第一个参数为网络输出值,FloatTensor类型,第二个参数为目标值,LongTensor类型。否则...原创 2018-09-24 20:49:53 · 7165 阅读 · 0 评论 -
pytorch源码编译遇到的问题
1. 编译到大概33%处,出现Failed to run 'bash ../tools/build_pytorch_libs.sh --use-cuda --use-nnpack nccl caffe2 libshm gloo c10d THD'解决:最新的pytorch源码编译要求CUDNN>5,升级CUDNN到6.0以上的版本即可。不需要更改/usr/local/cuda下边的c...原创 2018-09-25 11:37:13 · 4704 阅读 · 3 评论 -
pycharm远程调试服务器上的pytorch程序,Found no NVIDIA driver。
这个问题是在用pycharm远程调试功能时遇到的,当服务器上没有xforward功能时,pycharm专业版带有的远程调试功能是一个很好用的IDE调试解决方法。 程序运行时,torch.cuda.is_available()的值为False,而且模型无法使用GPU,当执行model.cuda()时出错。 执行print(torch.version.cuda...原创 2019-08-20 19:34:50 · 6524 阅读 · 0 评论