避坑宝典:PyTorch最简安装路径(含CUDA + VSCode + 中文手册)

这几天自己动手安装了一遍PyTorch,踩了无数坑,也试遍了各种方法。为了让后来者少走弯路,我把自己摸索出的最简便、最高效的实现路径整理出来,希望能让你一篇就够,轻松搞定PyTorch环境配置。

我的系统环境是Windows 10 + Intel CPU + Nvidia GPU,因此可以顺利启用CUDA加速,安装GPU版本的PyTorch。如果你是纯CPU环境,也可以参考本文,只需跳过CUDA相关步骤即可。下面直接进入正题,全程高能,简洁不啰嗦。

一、Anaconda ------ 打造隔离的Python环境

首先,安装Anaconda。它不仅是Python的包管理器,更是一个强大的环境管理工具,能帮我们把不同项目的依赖隔离开来,避免"版本冲突地狱"。

👉 官方下载链接

https://www.anaconda.com/download

下载安装包后,一路默认安装即可(记得勾选"Add Anaconda to my PATH environment variable"选项)。安装完成后,打开Anaconda Prompt(或终端),你会看到默认的base环境。

我们为PyTorch单独创建一个干净的环境,命令如下:

复制代码
conda create --name pytorch python=3.6

这里我选用Python 3.6,实际你也可以换成3.7或3.8,只要后续PyTorch支持即可。创建成功后,激活该环境:

创建成功之后,切换到PyTorch环境,

复制代码
conda activate pytorch

此时Python已默认安装。

二、CUDA ------ 开启GPU加速(可选)

如果你的电脑只有CPU,或者不打算用显卡加速,可以直接跳过这一步。

但如果你和我一样,拥有一块Nvidia显卡,并且想在训练模型时体验"起飞"的快感,那就需要安装CUDA。

打开下载页面选择你的操作系统、系统架构和安装包类型。根据显卡驱动版本,选择合适的CUDA版本(建议选10.2或11.x,因为PyTorch对这些版本支持最稳定)。

下载后运行安装程序,选择"自定义"安装,并确保勾选了"CUDA Development"相关组件。

安装完成后,可以在命令行输入nvcc --version验证是否成功。

三、PyTorch ------ 核心安装

在正式开始安装PyTorch之前,有一个极其关键的步骤:添加清华镜像源。如果不做这一步,默认从国外官方源下载,速度会"非常感人",很可能等到天荒地老。亲测有效,赶紧配置起来。

源搞定之后,进入,选择你的操作系统、包管理工具(Conda)、语言(Python)以及CUDA版本。网站会自动生成一行安装命令,例如:

复制命令,执行,安装。

安装完成之后,我们可以在Anaconda上面发现,多了一个环境:

四、VSCode ------ 丝滑的开发体验

环境装好了,总得有个趁手的编辑器。在众多Python开发工具中,我强推VSCode。它轻量、免费、插件丰富,而且对Anaconda环境支持得极为友好。

下载安装VSCode后,打开项目文件夹,点击左下角的Python解释器选择按钮,选择我们刚刚创建的pytorch环境中的Python解释器。此时,无论是代码补全、调试还是运行,都会自动使用这个隔离的环境,干净又优雅。

这就是实际界面,选择对应的解释器即可。

五、PyTorch中文手册 ------ 最佳入门教程

最后,为了让你的PyTorch学习之路更加顺畅,我强烈推荐一个开源项目------《PyTorch中文手册》。它内容简明扼要,从张量操作到模型训练,覆盖了绝大部分常用功能,非常适合作为案头参考书。

👉 GitHub地址:

https://github.com/apachecn/pytorch-doc-zh

相关推荐
2301_764150568 小时前
如何统计表单中已填写的特定类名输入框数量
jvm·数据库·python
2401_897190558 小时前
宝塔面板SSH提示连接被拒绝_检查服务器端口开关
jvm·数据库·python
2401_871696528 小时前
MySQL无法通过网络连接服务器_检查bind-address与访问权限
jvm·数据库·python
宸津-代码粉碎机8 小时前
Spring Boot 4.0虚拟线程实战续更预告:高阶技巧、监控排查与分布式场景落地指南
java·大数据·spring boot·分布式·后端·python
2401_887724508 小时前
SQL注入的安全架构设计_将数据库置于内网隔离区
jvm·数据库·python
Irene19918 小时前
Python zip() 函数详解
python
m0_678485458 小时前
如何配置文件描述符限制_limits.conf中Oracle用户配置
jvm·数据库·python
2401_835956818 小时前
HTML5中Canvas局部刷新区域重绘的算法优化
jvm·数据库·python
Irene19918 小时前
大数据开发场景中,Python 常用且易错易混淆的知识点总结(附:从实战角度梳理的 Python 知识体系)
大数据·python
weixin_408717778 小时前
如何导入带系统变量修改的SQL_确保SUPER权限并规避只读变量报错
jvm·数据库·python