windows安装anaconda、cuda、cudnn、pytorch全流程

本文详细介绍了在Windows系统中安装anaconda、cuda、cudnn以及pytorch的完整流程,包括环境变量配置、conda环境管理、换源设置以及如何在Pycharm中配置conda虚拟环境来运行Pytorch。同时,文中还提供了cuda与pytorch、显卡驱动的版本对应关系等关键信息。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

安装anaconda、cuda、cudnn、pytorch

Pycharm配置本地conda虚拟环境

  1. 安装anaconda,配置环境变量,修改环境和包安装位置,配置国内下载源地址channels

    win安装anaconda:官网下载安装包一路默认

    配置环境变量:此电脑右键属性>关于>高级系统设置>系统变量》Path》编辑》新建》输入anaconda路径

    修改安装位置:

     显示当前环境
     conda info --envs
     ​
     查看conda配置
     conda config --show
     ​
     增加环境路径
     conda config --add envs_dirs local_paths
     ​
     显示配置信息
     conda config --show
     ​
     删除环境路径
     conda config --remove envs_dirs local_paths

    换源修改channels:http://t.csdnimg.cn/y2R8D

    或者记事本打开:C:\Users\用户名\.condarc文件,替换为一下内容

     // 换源+改储存位置
     channels:
       - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
       - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
       - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
       - defaults
     show_channel_urls: true
     default_channels:
       - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
       - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
       - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
     custom_channels:
       conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
       msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
       bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
       menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
       pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
       pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
       simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
       deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/
     pkgs_dirs:
       - F:\anaconda\pkgs
     envs_dirs:
       - F:\anaconda\envs
     ​
     ​

  2. 打开anaconda prompt,安装环境

 // 创建新环境d2l,并安装python3.10
 conda create --name d2l python=3.10
 // 激活新环境
 conda activate d2l
  1. 检查显卡型号和支持cuda版本

     nvidia-smi
     高版本向下兼容,支持高版本就能跑低版本
    详见附录3 显卡驱动与cuda版本对应关系

    下载好cudatoolkit版本,直接conda install cudnn可以自动匹配cudnn版本

  2. 安装cuda/cudnn/pytorch

    附录2 cuda版本与pytorch版本对应关系

 conda install cudatoolkit=11.3.1
 ​
 conda install cudnn=8.2.1
 ​
 conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
 // 验证
 python
     import torch
     torch.cuda.is_available()
     // 返回ture
     exit()

// 或者 conda list|findstr torch // 返回torch版本

  1. pycharm运行环境改为conda的虚拟环境

右上角设置>项目名称《...》>Python解释器

右侧蓝色添加解释器>添加本地解释器

  1. 其他包,如opencv可以直接在pycharm中下载

附录1 anaconda基本命令

 http://t.csdnimg.cn/Y2CMD

附录2 cuda版本与pytorch版本对应关系

pytorch官网查看

 https://pytorch.org/get-started/previous-versions/

附录3 显卡驱动与cuda版本对应关系

 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

附录4 cuda与cudnn版本对应关系

 官网:
 https://developer.nvidia.cn/cudnn
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值