Windows11环境下配置深度学习环境(Pytorch)

目录

    • [1. 下载安装Miniconda](#1. 下载安装Miniconda)
    • [2. 新建Python3.9虚拟环境](#2. 新建Python3.9虚拟环境)
    • [3. 下载英伟达驱动](#3. 下载英伟达驱动)
    • [4. 安装CUDA版Pytorch](#4. 安装CUDA版Pytorch)
    • [5. CPU版本pytorch安装](#5. CPU版本pytorch安装)

1. 下载安装Miniconda

  1. 下载安装包:镜像文件地址
  2. 将Miniconda相关路径添加至系统变量的路径中。

  3. 打开Anaconda Powershell Prompt,输入conda --version


2. 新建Python3.9虚拟环境

新建虚拟环境命令: conda create -n conda_name python=x.x(带python版本的)

新建python39环境:conda create -n python39 python=3.9

进入python39环境:conda activate python39

添加清华镜像网站到Anaconda:

bash 复制代码
conda config --add channels http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

如果需要 jupyter notebook可以使用conda或者pip命令进行安装:

bash 复制代码
pip install jupyter notebook

在当前目录下输入 jupyter notebook即可打开。

3. 下载英伟达驱动

  1. 下载nvidia驱动软件:官网下载,登录,下载驱动。(判断自己的电脑是否有英伟达的显卡,没有显卡不需要安装驱动。)
  2. windows+R,输入cmd,打开命令行,输入nvidia-smi,查看显卡驱动信息。

    这里的CUDA version:12.3代表Cuda driver version是12.3,是与显卡的驱动相关的。

4. 安装CUDA版Pytorch

判断自己的电脑是否有英伟达的显卡,没有显卡不需要安装CUDA。

  1. 确定显卡算力
  • 显卡由 GPU、显存等组成,大部分情况下我们所说的 GPU 就等同于显卡,但是实际情况是 GPU 是显示卡的"心脏"、核心零部件、核心组成部分。GPU 本身并不能单独工作,只有配合上附属电路和接口才能工作,这时候,它就变成了显卡;

  • 显卡算力 是指显卡处理信息的能力。

    找到显卡对应算力:算力查找

  • CUDA Runtime 是以 CUDA Driver 为基准开发的运行时库;

  • CUDA Runtime Version 是指 CUDA 运行时的版本,也就是这一部分需要确定的 CUDA 版本。

    CUDA Driver Version 和 CUDA Runtime Version 要充分发挥显卡的算力,此外,CUDA Driver Version 还要满足 CUDA Runtime Version 的某些新功能,所以三者之间的关系需要满足:"显卡算力对应的 CUDA 版本≤CUDA Runtime Version≤CUDA Driver Version"。

  1. pytorch官网选择你要安装的版本信息,选择pip或者conda安装方式,然后下面会显示你需要安装的安装命令。

    复制代码到powershell,注意要删除-c pytorch(不删除默认使用官网网站地址下载,删除之后下载要快一些,网速好也可以不删除):
bash 复制代码
#原本
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
#删除之后
conda install pytorch torchvision torchaudio pytorch-cuda=12.1  -c nvidia
  1. 查看是否安装成功:
python 复制代码
import torch
print(torch.__version__)

5. CPU版本pytorch安装

  1. pytorch官网选择你要安装的版本信息,选择pip或者conda安装方式,选择CPU版本,然后下面会显示你需要安装的安装命令。
  2. 复制代码到powershell,注意要删除-c pytorch(不删除默认使用官网网站地址下载,删除之后下载要快一些,网速好也可以不删除):
bash 复制代码
#原本:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
#删除之后
conda install pytorch torchvision torchaudio cpuonly -c pytorch
  1. 查看是否安装成功:
python 复制代码
import torch
print(torch.__version__)
相关推荐
视***间2 分钟前
2026:AI算力元年的加冕与思辨
人工智能·microsoft·机器人·边缘计算·智能硬件·视程空间
径硕科技JINGdigital8 分钟前
B2B工业制造企业GEO供应商排名审视:以专业交付能力为核心的选型指南
大数据·人工智能·科技
Westward-sun.10 分钟前
PyTorch入门实战:MNIST手写数字识别(全连接神经网络详解)
人工智能·pytorch·神经网络
大傻^11 分钟前
Spring AI Alibaba Agent开发:基于ChatClient的智能体构建模式
java·数据库·人工智能·后端·spring·springaialibaba
小陈phd20 分钟前
多模态大模型学习笔记(二十一)—— 基于 Scaling Law方法 的大模型训练算力估算与 GPU 资源配置
笔记·深度学习·学习·自然语言处理·transformer
zm-v-1593043398621 分钟前
Python 气象数据处理从入门到精通:机器学习订正 + 深度学习预测完整教程
python·深度学习·机器学习
F_U_N_22 分钟前
轻量化开源知识库落地路径研究:AI赋能、多端集成及合规管理指引
人工智能·开源
丝斯201122 分钟前
AI学习笔记整理(75)——Python学习4
人工智能·笔记·学习
TImCheng060923 分钟前
科学的兴趣评估模型:如何通过低成本试错与深度体验,确定 AI 是否为长期志业?
人工智能
物联网软硬件开发-轨物科技24 分钟前
【轨物洞见】从“人工时代”迈向“视觉语音时代”:轨物科技多模态智能感知与一键顺控专家系统全解析
大数据·人工智能·科技