通过 pip 安装 pytorch
首先通过 python 官网将python安装,这一步先跳过了
直接进入主题 torch 安装
首先要进入 pytorch 官网(以给出)进入如下界面
点击 install 进入到配置选择的界面
Pytorch Build 这一行可供选择的是 稳定版(左侧)和 试用版 (右侧)
your OS 这里选择属于自己的电脑系统
Package 这里需要选择的是通过怎样的途径下载,本文以 pip 为例子进行安装操作
CUDA 在自己电脑上的系统信息中查看电脑的CUDA型号,这里实际上是确认电脑是否支持 GPU 计算
在选择好这些之后,在 Run this command 中会给出下载链接,将
pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
复制之后,win+R 输入 cmd 打开,将复制内容粘贴到这里就可以开始下载安装了
一般情况到这里就基本结束了,可以在命令行中进去python环境进行检测,这样说明安装成功,就可以利用 pytorch 来构建自己的深度学习模型了
但是 往往安装起来并不是那么简单
最常见的错误就是“找不到模块”“找不到安装包”等错误,而一般情况下可以通过以下方法解决,因为存在 python 和 pytorch 模块兼容问题,所以会存在python版本为3.7只能安装torch1.5 和torchvision0.6,
所以如果你安装的python版本是3.7或者之前的版本,那么在下载安装 pytorch 时就要考虑下载合适的版本
法1:可以在 pytorch 官网复制那个下载链接是将
pip install torch===1.7.0 torchvision===0.8.1 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
改为
pip install torch===1.5.0 torchvision===0.6.1 torchaudio===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
基本上就可以解决问题了
法2:通过如下链接下载对应的torch安装包来直接下载安装
https://download.pytorch.org/whl/torch_stable.html
更新1
问题:在调用 torch.from_numpy 时会出现 Module 'torch' has no 'from_numpy' memberpylint(no-member)的错误
解决:通过快捷键 Ctrl+Shift+P
打开vscode的命令面板,输入settings.json
并打开第一项,将如下内容复制进去并保存,即可解决问题
"python.linting.pylintArgs": [
"--errors-only",
"--generated-members=numpy.*, torch.*, cv2.*, cv.*"
]
更新2
问题:vs code 运行 torch.rand() 时出现 “Module 'torch' has no 'rand' memberpylint(no-member)” 的错误
解决:通过快捷键 Ctrl+Shift+P
打开vscode的命令面板,输入settings.json
并打开第一项,将如下内容复制进去并保存,即可解决问题
// whitelist torch to remove lint errors
"python.linting.pylintArgs": [
"--extension-pkg-whitelist=torch",
"--ignored-modules=torch",
"--ignored-classes=torch",
]
pip 清华园安装
-i https://pypi.tuna.tsinghua.edu.cn/simple
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!