pytorch常见问题的解决办法

pytorch问题随笔

Posted by JL on July 10, 2020

1. 在pycharm环境下,编译器选择提前创建好的cuda(gpu), import torch 报”ImportError: numpy.core.multiarray failed to import”

解决办法:

  1. 网上很多都是说remove old numpy version and install new version numpy at least numpy1.8.0 version.

  2. but, 我的情况是terminal 可以导入,也可以执行gpu 的相关程序,只是在pycharm 中不可以。
    想到说之前在cpu 环境下遇到类似的问题(具体什么问题忘记了,可见还是要记录,记录,记录,保存长期记忆的重要性)
    记得当时解决办法是将matplotlib 的版本降低了,从matplotlib 3.2.3 -> matplotlib 3.0.3
    今天的这个问题,又试验了一下老方法,依然管用(虽然问题不一样,但是答案一样,岂不快哉!!) ok!
    pip uninstall matplotlib

    ​ ` pip install matplotlib=3.0.3 `

2. 在用cuda的时候,自定义类,继承自nn.Module() 代码在运行过程中报:”RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 ‘self’ in call to _th_index_select”

解决办法:
1.来自stackoverflow 回答:
问题思路
2.针对我的问题,很简单:
model = SkipGramModel(100, 10).cuda()