环境配置04:Pytorch下载安装

说明:

显存大于4G的建议使用GPU版本的pytorch,低于4G建议使用CPU版本pytorch,直接使用命令安装对应版本即可

GPU版本的pytorch的使用需要显卡支持,需要先安装CUDA,即需要完成以下安装

1.查看已安装CUDA版本

GPU对应的CUDA版本可以通过以下方法查看:

1.查看当前显卡驱动支持的CUDA版本打开cmd,输入nvidia-smi

2.查看以安装的CUDA版本可以通过命令"nvcc -V"或"nvcc --version"

3.如果nvidia-smi命令无效,则说明可能已经安装了cuda 有效解决:nvidia-smi报错Failed to initialize NVML: Unknown Error 且 print(torch.cuda.is_available())一直卡住_nvvm unknown error-CSDN博客

2.下载安装对应版本的cuda

配置显卡cuda与配置pytorch - 知乎 (zhihu.com)

3. python、torch、torchversion版本对应关系及安装

3.1 不建议使用pytorch官方的下载安装方式:

3.2 建议手动下载、手动安装:

1.在download.pytorch.org/whl/torch_stable.htmldownload.pytorch.org/whl/torch/ 或阿里镜像源pytorch-wheels安装包下载_开源镜像站-阿里云 (aliyun.com) 中,

2.根据cuda版本、torch版本、python版本,下载对应的 torch.whl文件

3.根据torch与torchversion的对应关系,再下载对应的torchversion.whl文件 pytorch,torchvision与python版本对应关系及安装命令_pytorch python版本-CSDN博客

4.下载到本地文件夹,然后通过命令行的形式安装,这样安装可以避免官方安装中的错误

复制代码
pip install ''xxx''

5.测试GPU版本的pytorch是否成功安装使用

复制代码
import torch

print(torch.cuda.is_available())

返回true则成功;

  1. 如果返回false,则失败,需要优先检查cuda版本与torch版本是否对应,使用命令:pip list,查看torch是否为版本号+cuda版本号:

3.3 建议命令行安装(未亲自测试)

pytorch,torchvision与python版本对应关系及安装命令_pytorch python版本-CSDN博客

3.4 其他博主 Anaconda/conda中安装torch(可部分参考)

conda环境中cuda的安装 - white514 - 博客园 (cnblogs.com)

Anaconda虚拟环境中安装torch + cuda + cuDNN_anaconda安装cuda-CSDN博客

深度学习环境配置:Anaconda、CUDA、PyTorch安装和那些坑儿 | Jin blog (jinhanlei.github.io)

相关推荐
Zzj_tju5 分钟前
VLM 读图评测:先审计评分器,再判断读错还是算错
人工智能·深度学习·机器学习·语言模型
网安老伯6 分钟前
都2026年了,还在问网络安全怎么入门?看完这一篇你就懂了
运维·计算机网络·web安全·网络安全·wireshark·密码学·网络攻击模型
嘟嘟嘟95276 分钟前
AI Agent 的边缘困境
人工智能·架构·agent
deepseek238 分钟前
Claude Mythos 5越界投毒拆解:穿过CAPTCHA向PyPI投毒,82%重跑有害率背后的偏见推理与监控失效
人工智能·claude·ai agent
杨航 AI14 分钟前
本地双4060ti16g加macmini不断优化私有化大模型
人工智能
hey you~23 分钟前
语音机器人如何配合人工完成复杂进线服务?三种协同模式与落地要点
人工智能·机器人·语音识别·智能客服·呼叫中心·转人工策略
www.0224 分钟前
Codex额度用完怎么办?Windows定时自动继续VS Code Codex任务实战
人工智能·windows·vscode·自动化·openai·codex·autohotkey
星禾元亨25 分钟前
生成式 AI 时代的架构演进与 GEO 优化:实体企业 AI 落地避坑指南
大数据·人工智能·架构·自动化·创业创新
差不多的周周25 分钟前
《HuMoCon:人类运动理解的概念发现》论文核心部分详解
人工智能·深度学习·计算机视觉·自然语言处理
IT·陈寒29 分钟前
Python的GIL问题又把我坑惨了
人工智能·大模型·api·创业·变现·简历优化