Linux下cuda9.1安装Pytorch和Torchvision|简记

    技术2024-07-09  71

    Linux下cuda9.1安装Pytorch和Torchvision|简记


    PyToch和Torchvision对应版本 --官方安装命令


    我自己尝试了 该链接中对应的许多版本,然而大都报错:

    AssertionError: The NVIDIA driver on your system is too old (found version 9010). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.

    意思是我的cuda9.1版本不合适,让我 更新 cuda版本

    解决方法:

    如果是自己的服务器的话,我建议 直接 cuda版本 切换 到 9.0或者10.0 等其他版本,PyTorch 的支持会更好。坚持cuda9.1 不能变,那么 就要安装 torch1.1.0 和 torchvision0.3.0,这个对应版本亲测可用: pip install torch==1.1.0 和 pip install torchvision==0.3.0






    Processed: 0.021, SQL: 10