估计是用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文件最后即可。