基于特定的显卡和Ubuntu20.04 操作系统配置cuda + pytorch 环境
先总结下步骤。
建议有一台专门安装ubuntu20.04的电脑(双系统容易被windows升级损坏),要有支持cuda的nvidia显卡(上官网查询),否则用cpu版本的pytorch训练会很慢。
1、把手机上的抖音、吃鸡等卸载了!(遇到困难会无意识的打开浪费时间)
2、查看显卡支持的最高cuda版本,记录下来。(很重要!)
3、打开pytorch.org,查看对应cuda版本对应的pytorch。
4、记录pytorch和cuda版本,查看对应的python版本。
5、安装显卡驱动,重启。
6、安装cuda,测试之。
7、安装cudnn,测试之。
8、下载并安装anaconda。
9、在Anaconda中创建Python3.x环境,安装Keras。
10、安装正确版本的pytorch。
具体教程大家看这个。
https://zhuanlan.zhihu.com/p/51373519
这是cuda的官方安装教程。
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#pre-installation-actions