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

相关推荐
2401_88445415几秒前
CSS实现浮动图标与文本居中对齐_配合浮动与flex
jvm·数据库·python
OJAC111几秒前
考研失利空窗半年,他冲进智能体赛道
人工智能
如去几秒前
第七篇《AI重塑城市治理:从“被动响应”到“主动感知”的智慧升级》
人工智能
woxihuan1234561 分钟前
SQL统计分组内的所有数据唯一值_使用DISTINCT汇总
jvm·数据库·python
盼小辉丶2 分钟前
PyTorch强化学习实战(7)——表格学习与贝尔曼方程
pytorch·深度学习·强化学习
翼龙云_cloud2 分钟前
阿里云代理商:阿里云 GPU 服务器部署 DeepSeek V4指南
服务器·人工智能·阿里云·云计算·deepseek v4
m0_631529823 分钟前
CSS如何利用-disabled伪类禁用交互元素_通过灰度效果提升界面逻辑清晰度
jvm·数据库·python
程序员三明治3 分钟前
【AI】一文讲清 RAG:从大模型局限到企业级知识库落地流程
java·人工智能·后端·ai·大模型·llm·rag
dFObBIMmai4 分钟前
如何排查SQL存储过程内存溢出_优化大数据量临时表使用
jvm·数据库·python