Windows下PyTorch深度学习环境配置(GPU)

一:下载Anaconda

(路径最好全英文)

(下载好后,可以创建其他虚拟环境,因为是自己学习,所以先不放步骤,有需要者可以参考B站up我是土堆的视频)

二:利用 conda 或者 pip 安装 PyTorch

1. 确定自己安装的CUDA版本与GPU关系

1.确定显卡型号

(如图右上角,我是1050ti)

  1. 确定显卡算力 6.1

    (更多CUDA和GPU间的算力关系可参考https://zhuanlan.zhihu.com/p/544337083?utm_id=0)
  2. 确定CUDA Runtime

    (可以看到算力6.1对应的CUDA SDK,我选用目前最新的11.8)
  3. 我们看自己的驱动 CUDA Driver Version -- 12.2
    (打开cmd,输入nvidia-smi 就可看到cuda的版本,建议更新到最新)
    在-
    https://www.nvidia.cn/Download/index.aspx?lang=cn进行安装

    (将其中的信息选择自己电脑对应的型号,之后进行安装)

2. 安装GPU版本 PyTorch

打开pytorch官网

(本人如上)

将最下面那行复制,打开Anaconda prompt的窗口复制进去进行下载

(这里有镜像和pip的方法)

补pip:就是把第三行的conda换成Pip,再复制最后一行代码进行下载

三:验证

cpp 复制代码
conda env list  //显示环境
conda list  //显示环境下的包
conda activate 环境名  //切换环境
conda uninstall pytorch torchvison torchaudio cudatoolkit  //卸载那四个包

四:安装PyCharm并进行配置



PS:有更多疑问可以参考B站我是土堆

PS:他的博客

相关推荐
Rocky Ding*16 分钟前
一文读懂LLM Agent Skills 的运行时本质:从能力路由到渐进式披露
论文阅读·人工智能·深度学习·机器学习·aigc·ai-native·agent skills
大江东去浪淘尽千古风流人物1 小时前
【LoMa】局部特征匹配重访:从LoMa-B到旋转不变LoMa-R的架构与工程实践
开发语言·深度学习·计算机视觉·r语言·视觉定位·sfm·局部特征匹配
三玖诶1 小时前
DeepSeek Launcher:内置 Node.js,为什么首次启动仍需要网络?
windows·开源·deepseek
陈天伟教授1 小时前
具身数据采集黑话(4)
人工智能·windows·具身智能
江润舟1 小时前
万物 | 炼器 从零手搓工业级旋转目标检测网络 · 卷2 —— 计算图、梯度与反向传播(三)
人工智能·深度学习
zhurui_xiaozhuzaizai1 小时前
github上关于节省token的项目一览,上下文压缩,节省token,agent优化skill,热门项目
人工智能·深度学习·github
无巧不成书02181 小时前
Apache Tomcat 9 下载全指南(Windows,Linux,macOS):版本选型、国内镜像、SHA-512 与 PGP 完整性校验
windows·tomcat·apache
3Q工具箱 - BraggTroy2 小时前
Python + PyInstaller 实战:从零打造 10 个 Windows 桌面小工具(附全部踩坑记录与核心代码)
开发语言·windows·python
linnux领域2 小时前
eNSP 交换机与电脑不在一个网段?用一台 Windows 双网卡做路由,把实验环境彻底打通
windows·华为·电脑·ensp·静态路由·网络实验·ensp模拟器
维基框架2 小时前
PyTorch正在重构开源AI基础设施
人工智能·pytorch·重构