Windows10 python3.11+pytorch2.01+cuda11.8安装与配置

Windows10 python3.11+pytorch2.01+cuda11.8安装与配置

  • [1. 准备工作](#1. 准备工作)
  • [2. 安装过程](#2. 安装过程)
    • [2.1 Anaconda安装以及python环境配置](#2.1 Anaconda安装以及python环境配置)
    • [2.2 Cuda 安装](#2.2 Cuda 安装)
    • [2.3 cuDNN的安装](#2.3 cuDNN的安装)
    • [2.4 环境变量配置](#2.4 环境变量配置)
    • [2.5 离线安装pytorch](#2.5 离线安装pytorch)

1. 准备工作

本文提及的各软件下载版本仅供参考,大家根据自己需要去下载想要的版本,方法大同小异。

笔者之前也安装过pytorch和cuda,最近anaconda的环境出了点问题,干脆卸载了重装。以下是之前的安装参考链接,很多步骤都是类似的(这里显卡驱动就不重搞了,有需要的再看一下):
https://blog.csdn.net/tzr0725/article/details/115351082

pytorch与cuda的搭配可参考如下链接:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

管理Python虚拟环境,Anaconda下载(习惯性从清华开源镜像下载):
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

cuda下载:https://developer.nvidia.com/rdp/cudnn-download

cudnn下载:https://developer.nvidia.com/rdp/cudnn-download

pytorch离线下载链接:
https://download.pytorch.org/whl/torch_stable.html

笔者整理好了对应的版本,有需要的自取:

链接:https://pan.baidu.com/s/1TBB_hz-tY3StOXUHPEwxkg

提取码:wzfg

2. 安装过程

2.1 Anaconda安装以及python环境配置

先安装anaconda

这里笔者忘了点添加变量,最好还是选上


检查是否安装成功,在开始菜单,点击Anaconda Prompt

打开后,左边有(base),即安装成功

创建虚拟环境

powershell 复制代码
conda create -n pytorch python=3.11

查看安装的虚拟环境

激活安装的虚拟环境

利用pip工具安装一些基础第三方库,比如numpy

powershell 复制代码
pip3 install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple/

2.2 Cuda 安装


提示没有安装visualstudio,按照提供的链接下载安装

安装visual studio

继续安装cuda

2.3 cuDNN的安装

对于cuDNN直接将其解开压缩包,然后需要将bin,include,lib中的文件复制粘贴到cuda的文件夹下:

powershell 复制代码
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

2.4 环境变量配置

此电脑右键-属性-高级系统设置-环境变量

验证是否安装成功

2.5 离线安装pytorch

利用终端进入下载好离线文件的路径,输入pip3 install 文件名即可进行安装

安装完成后再将官网中生成的指令在终端输入,因为安装过torch,所以只会安装torchvision和torchaudio,接下来就会比较快

官网链接:
https://pytorch.org/get-started/locally/

powershell 复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

验证是否安装成功:

大功告成...

相关推荐
靴子学长29 分钟前
Decoder only 架构下 - KV cache 的理解
pytorch·深度学习·算法·大模型·kv
智算菩萨29 分钟前
【OpenGL】10 完整游戏开发实战:基于OpenGL的2D/3D游戏框架、物理引擎集成与AI辅助编程指南
人工智能·python·游戏·3d·矩阵·pygame·opengl
jason成都1 小时前
IoT 设备监控系统实战:基于 EMQX 的 MQTT 连接监控与数据格式指纹识别
开发语言·python
愤豆2 小时前
05-Java语言核心-语法特性--模块化系统详解
java·开发语言·python
独隅2 小时前
PyTorch 模型性能优化:图像分类与 NLP 模型实战指南
pytorch·性能优化·分类
AI-Ming2 小时前
程序员转行学习 AI 大模型: 踩坑记录:服务器内存不够,程序被killed
服务器·人工智能·python·gpt·深度学习·学习·agi
Narrastory2 小时前
明日香 - Pytorch 快速入门保姆级教程(九)
人工智能·pytorch·深度学习
2401_873544922 小时前
使用Python处理计算机图形学(PIL/Pillow)
jvm·数据库·python
njidf3 小时前
自动化机器学习(AutoML)库TPOT使用指南
jvm·数据库·python
只与明月听3 小时前
RAG深入学习之向量数据库
前端·人工智能·python