Advertisement

Pytorch的GPU driver is too old

阅读量:
复制代码
    RuntimeError: The NVIDIA driver on your system is too old (found version 10010). 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.

在服务器上运行pytorch版本的训练模型时,出现了这样的错误。以为是服务器上的cuda版本过于老旧。后面看过大佬的解说之后才知道是cuda版本跟pytorch的版本没有适配
首先远程服务器查看cuda的版本

复制代码
    cat /usr/local/cuda/version.txt
    #这里我的版本是CUDA Version 10.1.243

在这里插入图片描述
找了一张大佬总结的图可以根据这个查找对应版本这里贴几个自己常用的,如果没有适合自己的可以去官网按需自取
cuda版本对应pytorch安装

复制代码
    # CUDA 9.2
    conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=9.2 -c pytorch
    
    # CUDA 10.1
    conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch
    
    # CUDA 10.2
    conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch
    
    # CPU Only
    conda install pytorch==1.6.0 torchvision==0.7.0 cpuonly -c pytorch

可能由于网络原因利用conda 安装包的时候卡住,我利用ctrl+c中断进程后,再次利用conda install 命令安装,发现一直出现安装失败,提示"段错误(核心已转储)",且再利用 conda install 命令安装任何包时都安装失败。
运行

复制代码
    conda clean -a

重新运行conda install 安装

全部评论 (0)

还没有任何评论哟~