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

相关推荐
weiwuxian12 小时前
🚀 拒绝“人工智障”:如何写出一份高质量的 agent.md?
人工智能
黑客思维者13 小时前
智能配电系统用户敏感数据脱敏详细设计:从静态遮盖到动态策略
c++·python·嵌入式系统·数据脱敏·智能配电系统
陈鋆13 小时前
Langchain-Chatchat[四、RAG对话流程代码解析]
开发语言·python·langchain
ServBay13 小时前
Django 6.0 发布,新增原生任务队列与 CSP 支持
后端·python·django
Yeats_Liao13 小时前
CANN Samples(十一):媒体处理接口V1与V2的取舍与迁移
人工智能·媒体
β添砖java13 小时前
python第一阶段第九章异常、模块、包
开发语言·python
2501_9419820513 小时前
企业微信Python SDK:高效群发消息实战
开发语言·python·企业微信
汤姆yu13 小时前
基于srpingboot心情治愈调整系统
人工智能
国科安芯13 小时前
航天医疗领域AS32S601芯片的性能分析与适配性探讨
大数据·网络·人工智能·单片机·嵌入式硬件·fpga开发·性能优化
咚咚王者13 小时前
人工智能之数据分析 Pandas:第五章 文件处理
人工智能·数据分析·pandas