总结深度学校ubuntu环境配置
目录
- 0.前提工作
- 0. nvidia驱动安装
- 1.CudaTookits安装
- 2. cuDNN安装
- 3.pytorch安装
0.前提工作
安装顺序:nvidia驱动->cudaTookits->cudnn->TensorRT->pytorch->tensorflow
版本匹配方式:
- 1.先决定tensorflow支持的版本
进入他官tensorflow官网之后右上角将语言选择为English。因为英文版支持版本比中文版更多更新
Version | Python version | Compiler | Build tools | cuDNN | CUDA |
---|---|---|---|---|---|
tensorflow-2.16.1 | 3.9-3.12 | Clang 17.0.6 | Bazel 6.5.0 | 8.9 | 12.3 |
-
2.决定cudaTookits版本 - CUDA12.3
-
3.决定nvidia驱动版本
0. nvidia驱动安装
参考:
1.超详细ubuntu20.04/22.04安装nvidia驱动/CUDA/cudnn
2.Python-Pytorch开发环境使用总结
nvidia显卡驱动安装方式有三种:使用ubuntu附加驱动的方式;使用命令行方式安装;使用.run文件的方式进行安装
ubutnu上推荐使用命令行方式安装
sudo add-apt-repository ppa:graphics-drivers/ppa # 加入官方ppa源
sudo apt update # 检查软件包更新列表
apt list --upgradable # 查看可更新的软件包列表
sudo apt upgrade