热门标签
热门文章
- 1[Rust 入门]Rust 结构体总结_rust 结构体对齐
- 2强化学习入门DQN详解_dqn神经网路训练参数训练过程
- 3图像相似性搜索思路_图像寻找相似物
- 4Python pandas 操作 excel 详解_pandas处理excel_pd输出成excel
- 5前端学习之vue+element-ui电商项目（四）角色列表_element ui 角色 用户
- 6JWT加入注解实现权限认证_jwtpermission
- 7LLM微调工具TorchTune易用性高于一切
- 8深度探索语义分割：DeepLabv3_MobileNetv2 —— 强力的轻量化选择
- 9AI视觉相关的预训练模型、公开数据集_ai视觉模型 训练
- 10开源ffmpeg（四）——使用SDL进行音频播放上（编译篇）_vlc 音频输出调用的是sdl吗
当前位置: article > 正文
日常随笔1：配置pytorch
作者：空白诗007 | 2024-08-17 09:00:48
赞
踩
日常随笔1：配置pytorch
踩到的坑
1.安装pytorch时用的不是官方的源，换成了清华源，然后运行的时候无法支持cuda；换成官网的可以。
2.numpy版本为2.0，torch有些功能不兼容，可以降低版本。我的python为3.9，numpy版本改为了1.26，实测可用。
pip install numpy==1.26| Python 2.7的最后一个版本 | NumPy 1.16.5 |
| Python 3.5的最后一个版本 | NumPy 1.19.5 |
| Python 3.6及以上版本 | NumPy 1.20.3 |
安装步骤
1.安装anaconda
创建conda虚拟环境
2.查看可安装的cuda版本并安装（注意GPU的算力，我的算力是457.63，只能安装11.1以下的CUDA版本）
3.去官网下载符合cuda版本的pytorch版本
安装完毕。
在pycharm中调用torch
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/空白诗007/article/detail/992064
推荐阅读
相关标签
