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")

相关推荐
程序员小远1 分钟前
Web自动化测试详解
自动化测试·软件测试·python·selenium·测试工具·职场和发展·测试用例
lomocode1 分钟前
大模型本地部署与预热全攻略:让首次响应速度提升 5 倍
人工智能
生信大表哥6 分钟前
如何在服务器上使用 Gemini 3 进行生信分析:从入门到进阶
linux·人工智能·语言模型·数信院生信服务器·生信云服务器
某林2127 分钟前
SLAM 建图系统配置与启动架构
人工智能·stm32·单片机·嵌入式硬件·算法
红石程序员8 分钟前
Python环境管理
开发语言·python
cooldream20099 分钟前
深入理解 Cursor 规则体系
人工智能·cursor
Chennnng10 分钟前
关于python版本,显卡版本,torch版本之间的问题
开发语言·python
AI浩16 分钟前
【Qwen3-VL-4B-Instruct实战】推理图片和视频、加速
人工智能
腾飞开源22 分钟前
26_Spring AI 干货笔记之 OCI GenAI Cohere 聊天
人工智能·自动配置·依赖管理·springai·聊天模型·运行时选项·oci cohere
战南诚22 分钟前
python序列化-dumps的妙用
python