深度学习-Ubuntu系统预装环境¶
本文预装软件和环境为ubuntu22.04系统，其他Ubuntu系统暂只提供纯净版
常规软件
| 已安装 | |||
|---|---|---|---|
PyCharm 社区版 |
Anaconda |
Docker |
Chrome |
VsCode |
Jupyter notebook |
Git |
百度网盘 |
CUDA
- 系统默认已安装显卡驱动、
cuda12.4版本与cudnn8.9.7版本，除非特殊需要您应该优先使用Anaconda虚拟环境，虚拟环境会有独立的cuda与cudnn版本； - 如果没有二次编译代码的需求，正常情况下不需要单独安装CUDA/cuDNN;
- 如果您对
驱动和cuda设置不是很熟悉，请勿随意更改以免造成系统无法使用；
PyTorch¶
- 所有的环境都安装在
Anaconda虚拟环境中，并已经默认安装了多个版本，您可以根据您的需要选择适合的版本。 - 点击查看Anaconda使用说明
| 环境名称 | Python | PyTorch | CUDA | cuDNN |
|---|---|---|---|---|
| py38-torch112-cuda113 | 3.8 | 1.12.0 | 11.3 | 8.2 |
| py39-torch201-cuda118 | 3.9 | 2.0.1 | 11.8 | 8.9 |
| py311-torch222-cuda121 | 3.11 | 2.2.2 | 12.1 | 8.9 |
YOLO¶
系统默认已经安装yolov5版本
| 环境名称 | Python | PyTorch | Yolo | CUDA | cuDNN |
|---|---|---|---|---|---|
| py39-yolov5-torch201-cuda118 | 3.9 | 2.0.1 | 5 | 11.8 | 8.9 |
TensorFlow¶
| 环境名称 | Python | TensorFlow | CUDA | cuDNN |
|---|---|---|---|---|
| py38-tf25-cuda113 | 3.8 | 2.5.0 | 11.3 | 8.1 |
5090¶
5090需要使用PyTorch 2.7.0以上版本， 5090机型已经默认配置
| 环境名称 | Python | PyTorch | CUDA | cuDNN |
|---|---|---|---|---|
| py311-torch270-cuda128 | 3.11 | 2.7.0 | 11.8 | 9.0 |