pytorch gpu安装

cuda

https://blog.csdn.net/qq_51570094/article/details/124148671

https://blog.csdn.net/zxdd2018/article/details/127705627

cudnn

https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#installlinux-tar

更改cudnn

保证文件目录中只有一个解压后的cudnn文件夹

复制代码
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

pytorch版本匹配问题

nvcc --version和nvidia-smi的版本可能會不一樣,前者是runtime api對應的版本,後者是driver api的版本。

pytorch版本通常需要和runtime的版本匹配,driver的版本可以向下兼容。

相关推荐
张较瘦_1 分钟前
[论文阅读] AI + 软件工程(Debug)| 告别 “猜 bug”:TreeMind 用 LLM+MCTS 破解 Android 不完整报告复现难题
论文阅读·人工智能·bug
万粉变现经纪人7 分钟前
如何解决 pip install 安装报错 ImportError: cannot import name ‘xxx’ from ‘yyy’ 问题
python·selenium·测试工具·flask·scikit-learn·fastapi·pip
深栈8 分钟前
机器学习:线性回归
人工智能·pytorch·python·机器学习·线性回归·sklearn
AI视觉网奇27 分钟前
虚拟机安装 网络问题
人工智能·虚拟机
云澈ovo37 分钟前
FP16混合精度训练:Stable Diffusion生成速度提升300%的硬件配置方案
人工智能·机器学习·stable diffusion
zzywxc78742 分钟前
AI行业应用:金融、医疗、教育、制造业的落地实践与技术创新
人工智能·机器学习·金融·自动化·prompt·ai编程·xcode
简简单单做算法1 小时前
基于遗传优化的LSTM-Attention一维时间序列预测算法matlab仿真
人工智能·lstm·时间序列预测·lstm-attention·遗传优化
C++chaofan1 小时前
项目中为AI添加对话记忆
java·数据结构·人工智能·redis·缓存·个人开发·caffeine
simon_skywalker1 小时前
第三章 字典与集合
python
Elastic 中国社区官方博客1 小时前
CI/CD 流水线与 agentic AI:如何创建自我纠正的 monorepos
大数据·运维·数据库·人工智能·搜索引擎·ci/cd·全文检索