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 小时前
程序员在线接单
java·jvm·后端·python·eclipse·tomcat·dubbo
搞笑的秀儿1 小时前
信息新技术
大数据·人工智能·物联网·云计算·区块链
云天徽上1 小时前
【PaddleOCR】OCR表格识别数据集介绍,包含PubTabNet、好未来表格识别、WTW中文场景表格等数据,持续更新中......
python·ocr·文字识别·表格识别·paddleocr·pp-ocrv5
阿里云大数据AI技术1 小时前
OpenSearch 视频 RAG 实践
数据库·人工智能·llm
你怎么知道我是队长2 小时前
python-input内置函数
开发语言·python
XMAIPC_Robot2 小时前
基于ARM+FPGA的光栅尺精密位移加速度测试解决方案
arm开发·人工智能·fpga开发·自动化·边缘计算
加油吧zkf2 小时前
YOLO目标检测数据集类别:分类与应用
人工智能·计算机视觉·目标跟踪
叹一曲当时只道是寻常2 小时前
Python实现优雅的目录结构打印工具
python
Blossom.1182 小时前
机器学习在智能制造业中的应用:质量检测与设备故障预测
人工智能·深度学习·神经网络·机器学习·机器人·tensorflow·sklearn
天天扭码2 小时前
AI时代,前端如何处理大模型返回的多模态数据?
前端·人工智能·面试