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 分钟前
复杂任务的分布式智能解决方案
人工智能·分布式
DashVector7 分钟前
通义深度搜索-上传文件
人工智能·深度学习·阿里云·ai·深度优先
龙山云仓7 分钟前
No131:AI中国故事-对话荀子——性恶论与AI约束:礼法并用、化性起伪与算法治理
大数据·人工智能·深度学习·算法·机器学习
半路_出家ren7 分钟前
1.古典密码概述
python·网络安全·密码学·古典密码·加密方式
柠萌f10 分钟前
從靈感到跑量一站完成?
人工智能
CJenny36 分钟前
Claude Code常用操作和使用方法
人工智能·python
2501_9413331039 分钟前
铁路轨道部件识别与分类_YOLO13与BAMConv改进模型实现_1
人工智能·分类·数据挖掘
九河云40 分钟前
纺织印染“数字色差仪”:光谱+AI模型一次调色成功省染料12%
大数据·人工智能·安全·机器学习·数字化转型
星火开发设计1 小时前
二维数组:矩阵存储与多维数组的内存布局
开发语言·c++·人工智能·算法·矩阵·函数·知识
2501_911067661 小时前
光能赋能,步步生 “电”!叁仟智慧路灯杆 + 太阳能地砖,解锁智慧城市新范式
人工智能·智慧城市