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

验证是否安装成功:

大功告成...

相关推荐
凤枭香37 分钟前
Python OpenCV 傅里叶变换
开发语言·图像处理·python·opencv
测试杂货铺44 分钟前
外包干了2年,快要废了。。
自动化测试·软件测试·python·功能测试·测试工具·面试·职场和发展
艾派森1 小时前
大数据分析案例-基于随机森林算法的智能手机价格预测模型
人工智能·python·随机森林·机器学习·数据挖掘
SpikeKing1 小时前
LLM - 使用 LLaMA-Factory 微调大模型 环境配置与训练推理 教程 (1)
人工智能·llm·大语言模型·llama·环境配置·llamafactory·训练框架
小码的头发丝、1 小时前
Django中ListView 和 DetailView类的区别
数据库·python·django
Chef_Chen2 小时前
从0开始机器学习--Day17--神经网络反向传播作业
python·神经网络·机器学习
千澜空2 小时前
celery在django项目中实现并发任务和定时任务
python·django·celery·定时任务·异步任务
斯凯利.瑞恩2 小时前
Python决策树、随机森林、朴素贝叶斯、KNN(K-最近邻居)分类分析银行拉新活动挖掘潜在贷款客户附数据代码
python·决策树·随机森林
yannan201903133 小时前
【算法】(Python)动态规划
python·算法·动态规划
蒙娜丽宁3 小时前
《Python OpenCV从菜鸟到高手》——零基础进阶,开启图像处理与计算机视觉的大门!
python·opencv·计算机视觉