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

相关推荐
且慢.5897 分钟前
命令行的学习使用技巧
python
共绩算力12 分钟前
OpenAI Whisper 语音识别模型:技术与应用全面分析
人工智能·whisper·语音识别·共绩算力
海琴烟Sunshine21 分钟前
leetcode 66.加一 python
python·算法·leetcode
工藤学编程23 分钟前
零基础学AI大模型之Stream流式输出实战
人工智能
不良人龍木木36 分钟前
机器学习-常用库
人工智能·机器学习
罗橙7号40 分钟前
【pyTorch】关于PyTorch的高级索引机制理解
人工智能·pytorch·python
B站计算机毕业设计之家40 分钟前
智能监控项目:Python 多目标检测系统 目标检测 目标跟踪(YOLOv8+ByteTrack 监控/交通 源码+文档)✅
python·yolo·目标检测·目标跟踪·智慧交通·交通·多目标检测
rengang661 小时前
09-随机森林:介绍集成学习中通过多决策树提升性能的算法
人工智能·算法·随机森林·机器学习·集成学习
江上月5131 小时前
django与vue3的对接流程详解(下)
后端·python·django
nightunderblackcat1 小时前
四大名著智能可视化推演平台
前端·网络·爬虫·python·状态模式