基于特定的显卡和Ubuntu20.04 操作系统配置cuda+pytorch环境(适用于初学者)

基于特定的显卡和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