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

验证是否安装成功:

大功告成...

相关推荐
电子_咸鱼5 小时前
动态规划经典题解:单词拆分(LeetCode 139)
java·数据结构·python·算法·leetcode·线性回归·动态规划
风轻扬77712 小时前
SQLAlchemy2.0使用
python·sqlalchemy·orm框架
潘帕斯的雄鹰12 小时前
直观理解注意力机制
python·transformer·注意力机制·自注意力机制
打酱油程序员12 小时前
Python数据挖掘详细学习指南
python
天天进步201514 小时前
Python全栈项目:结合Puppeteer和AI模型操作浏览器
开发语言·人工智能·python
闲人编程15 小时前
用Python识别图片中的文字(Tesseract OCR)
开发语言·python·ocr·识图·codecapsule
盘古开天166615 小时前
从零开始:如何搭建你的第一个简单的Flask网站
后端·python·flask
二进制星轨16 小时前
Transofrmer架构详解与PyTorch实现(附代码讲解)
人工智能·pytorch·python
生而为虫16 小时前
02.第一个Python程序
开发语言·python
视觉AI16 小时前
如何查看 Linux 下正在运行的 Python 程序是哪一个
linux·人工智能·python