Linux Anaconda虚拟环境 GPU Pytorch 安装配置流程问题总结

Linux-Anaconda虚拟环境 安装配置GPU版本Pytorch 流程+遇到与解决问题小结

• 遇到问题

  • 流程总结

  • • 1.确认当前系统所使用的cuda版本信息
    • 2.编辑.bashrc配置文件，添加相应的cuda环境变量路径
    • 3.建立一个专用于pytorch-gpu的虚拟环境
    • 4.前往pytorch官方网站，根据所安装的cuda版本选择对应的gpu版本进行安装
    • 5.完成安装后进行相关测试以验证是否正常

  • 遇到各种问题、Bug小结

  • • 1.通过三种不同方式（nvcc -V、cat /usr/local/cuda/version.txt、nvidia-smi）查看cuda版本时得到的结果并不一致
    • 2.在系统中存在多个cuda版本的情况下，使用nvcc –V查看的版本可能与实际软链接指向的cuda版本不匹配
    • 3.在安装pytorch过程中若未指定具体的pytorch和torchvision版本号，则默认会安装torch1.4.0，该版本与当前cuda10.0存在兼容性问题！
    • 4.RuntimeError("{} is a zip archive (did you mean to use torch.jit.load()?)".format(f.name)