0
点赞
收藏
分享

微信扫一扫

如何在conda虚拟环境中安装所需版本的cuda和cudnn

穆风1818 2022-05-02 阅读 76

如何在conda虚拟环境中安装所需版本的cuda和cudnn

创建好虚拟环境后,激活环境,

(1)在环境中安装cuda

conda install cudatoolkit=你所需要的版本(例如 cudatoolkit=10.1
 
 

    若安装不成功则使用镜像源安装:

    conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
    
     
     

      (2)在环境中安装cudnn

      conda install cudnn=你所需要的版本(例如cudnn=7.6.5
       
       

        若安装不成功则使用镜像源安装:

        conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
        
         
         

          如果在安装的时候没有写版本,则会自己匹配与cuda版本相匹配的版本。

          举报

          相关推荐

          0 条评论