pytorch安装问题


目录
  • 1.版本
  • 2.pytorch调试中出现的Module 'torch' has no 'zero' member如何解决
  • 3.No module named 'numpy.core._multiarray_umath'
  • 4.Torch not compiled with CUDA enabled

pytorch是学术科研上经常使用的机器学习框架,而tensorflow则更多应用在机器学习的工业领域。

1.版本

torch 0.4.1对应torchvision 0.2.1

pip install torchvision==0.2.1

2.pytorch调试中出现的Module 'torch' has no 'zero' member如何解决

在设置中搜索python.linting.pylintPath
将原位置的pylink跟换成——anaconda文件夹中寻找pylint应用程序的安装位置

3.No module named 'numpy.core._multiarray_umath'

更新Numpy版本

4.Torch not compiled with CUDA enabled

查看电脑CUDA版本,下载对应torch