Pytorch CUDA11.4版本匹配

1.查看CUDA,并查找对应的torch的版本

首先判断自己是否适合使用GPU版本的torch,打开自己的cmd输入

复制代码
NVIDIA-smi

CUDA11.4版本比较特殊的版本,可以用CUDA11.3版本

2.下载orch、torchvision、torchaudio(三者版本要相关)

知道pytorch可以下载对应CUDA11.3版本的,可以从pytorch官方网站[3]去下载

复制代码
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

从torch.cuda.is_available()返回false 一路走到这里,刚接触pytorch,暂时就这样了

References:

1\] [torch.cuda.is_available()返回false------解决办法_cuda available false-CSDN博客](https://blog.csdn.net/qq_46126258/article/details/112708781 "torch.cuda.is_available()返回false——解决办法_cuda available false-CSDN博客") \[2\] [CUDA11.4配置pytorch、pytorchvision_cuda pytorch_人工智能之浪潮的博客-CSDN博客](https://blog.csdn.net/mago2015/article/details/131295825 "CUDA11.4配置pytorch、pytorchvision_cuda pytorch_人工智能之浪潮的博客-CSDN博客") \[3\] [Previous PyTorch Versions \| PyTorch](https://pytorch.org/get-started/previous-versions/ "Previous PyTorch Versions | PyTorch")

相关推荐
小二·2 小时前
Python Web 开发进阶实战:性能压测与调优 —— Locust + Prometheus + Grafana 构建高并发可观测系统
前端·python·prometheus
leo__5202 小时前
基于MATLAB的交互式多模型跟踪算法(IMM)实现
人工智能·算法·matlab
脑极体3 小时前
云厂商的AI决战
人工智能
njsgcs3 小时前
NVIDIA NitroGen 是强化学习还是llm
人工智能
七牛云行业应用3 小时前
重构实录:我删了 5 家大模型 SDK,只留了 OpenAI 标准库
python·系统架构·大模型·aigc·deepseek
知乎的哥廷根数学学派3 小时前
基于多模态特征融合和可解释性深度学习的工业压缩机异常分类与预测性维护智能诊断(Python)
网络·人工智能·pytorch·python·深度学习·机器学习·分类
mantch3 小时前
Nano Banana进行AI绘画中文总是糊?一招可重新渲染,清晰到可直接汇报
人工智能·aigc
编程小白_正在努力中4 小时前
第1章 机器学习基础
人工智能·机器学习
一人の梅雨4 小时前
亚马逊SP-API商品详情接口轻量化实战:合规与商业价值提取指南
python
wyw00004 小时前
目标检测之SSD
人工智能·目标检测·计算机视觉