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

相关推荐
季布,6 分钟前
本地Windows测试:钉钉群消息/文件传输到Python服务(完整教程)
windows·python·钉钉
2501_9361460411 分钟前
目标检测论文解读复现之六基于RetinaNet的考拉检测方法
人工智能·目标检测·计算机视觉
zm-v-1593043398612 分钟前
最新AI-Python自然科学领域机器学习与深度学习技术
人工智能·python·机器学习
qwerasda12385216 分钟前
Mask-RCNN右转交通标志识别训练与优化
python
郝学胜-神的一滴24 分钟前
何友院士《人工智能发展前沿》全景解读:从理论基石到产业变革
人工智能·python·深度学习·算法·机器学习
2401_8403652128 分钟前
cuda-gdb Could not find CUDA Debugger back-end.
人工智能
苍何fly36 分钟前
首个国产芯片训练的多模态 SOTA 模型,已免费开源!
人工智能·经验分享
2401_8414956438 分钟前
具身智能:从理论到现实,人工智能的下一场革命
人工智能·算法·机器人·硬件·具身智能·通用智能·专用智能
方见华Richard1 小时前
对话量子场论:语言如何产生认知粒子V0.3
人工智能·交互·学习方法·原型模式·空间计算
wfeqhfxz25887821 小时前
基于YOLO12-A2C2f-DFFN-DYT-Mona的铁件部件状态识别与分类系统_1
人工智能·分类·数据挖掘