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的版本可以向下兼容。

相关推荐
爬虫程序猿43 分钟前
用 Python 给京东商品详情做“全身 CT”——可量产、可扩展的爬虫实战
开发语言·爬虫·python
诗句藏于尽头1 小时前
自动签到之实现掘金模拟签到
python·1024程序员节
FlagOS智算系统软件栈2 小时前
与创新者同频!与FlagOS共赴开源之约
人工智能·ai·开源
蓝纹绿茶3 小时前
bash:**:pip:***python: 错误的解释器: 没有那个文件或目录
开发语言·python·pip
加油吧zkf3 小时前
循环神经网络 RNN:从时间序列到自然语言的秘密武器
人工智能·rnn·自然语言处理
START_GAME3 小时前
深度学习Diffusers:用 DiffusionPipeline 实现图像生成
开发语言·python·深度学习
Deamon Tree4 小时前
后端开发常用Linux命令
linux·运维·python
koo3645 小时前
李宏毅机器学习笔记30
人工智能·笔记·机器学习
长桥夜波5 小时前
机器学习日报02
人工智能·机器学习·neo4j
Elastic 中国社区官方博客5 小时前
介绍 Elastic 的 Agent Builder - 9.2
大数据·运维·人工智能·elasticsearch·搜索引擎·ai·全文检索