Win10 Anaconda 下配置CUDA、cudnn和pytorch

1.  Anaconda 安装

(1)从官网下载文件,按照提示安装即可,建议不要安装到C盘

   官网地址:https://www.anaconda.com/download

    重要的是配置下图所示的环境变量。依次点开  1 设置---> 2高级系统设置(对话框右侧)---> 2高级系统设置(对话框右侧)---> 3环境变量(对话框右下侧)---> 4 双击 Path ---> 5 新建添加三个环境变量。   

(2)安装成功验证:

打开Anaconda3下面的Prompt,输入命令:conda --version,显示的是自己下载的版本则安装正确。

2.CUDA Pytorch

(1)查看本机GPU环境

     此电脑(通过设置---个性化---主题----(右侧)桌面图标设置---选择 计算机)右键打开管理,查看驱动器版本(蓝线部分为驱动版本,目前的为556.12版本),这个版本决定了后续可安装的cuda等的版本。 

 如果没有相关的驱动信息,可以按照显卡型号,从NVIDIA GeForce 驱动程序的官网下载,网址为:  https://developer.nvidia.com/cuda-toolkit-archive

(2)准备工作(非常重要

       查看CUDA和驱动版本的对应关系:

       从官网  https://docs.nvidia.com/deploy/cuda-compatibility/index.html  上查看CUDA版本和驱动版本的对应关系。注意:一定要确保驱动和CUDA能够对应上。

                 

       pytorch与python的对应关系:从下面网址中查看

GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision

       pytorch 与cuda的版本对应关系:

      建议依据cuda版本选择合适的pytorch版本,参考官网:https://pytorch.org/get-started/previous-versions/

(3)安装 pytorch cudnn  cuda

     在Ananconda Prompt使用如下命令进行安装(注意黄色部分的版本是需要依据自己机器进行选择的部分):

           #conda create -n pytorch21 python=3.9  ##建立anaconda虚拟环境

           #conda activate pytorch21

         #安装pytorch、cuda、cudnn,命令来源于上图网址,选择适用于自己机器的版本号

       #conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia  

(4)验证安装结果

        在Ananconda Prompt使用如下命令进行验证

  • conda activate pytorch21
  • python
  • import torch
  • print(torch.cuda.is_available())  #返回True则说明已经安装了cuda
  • #判断是否安装了cuDNN
  • from torch.backends import  cudnn
  • print(cudnn.is_available())  #返回True则说明已经安装了cuDNN
  • #判断cuda版本
  • print(torch.__version__)
  • print(torch.version.cuda)
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值