import torchvision时出现问题:段错误(核心已转储)

估计是用deb安装cuda9.1后需要手动调整路径,导致安装完pytorch和torchvision后import时一直出现段错误。

参考了https://github.com/pytorch/pytorch/issues/4101,在最后一个评论有人说直接把nvidia的动态链接库的安装路径放到变量 LD_LIBRARY_PATH 里就好了。

于是直接在终端输入:

export LD_LIBRARY_PATH=/usr/lib/nvidia-387  #安装的驱动是387

再import torchvision

问题可以解决。

但是每次重启都要输入一次,比较麻烦,所以把这句话放到.bashrc文件最后即可。

你可能感兴趣的:(import torchvision时出现问题:段错误(核心已转储))