Ubuntu18.04.5+GTX

GTX"/>

Ubuntu18.04.5+GTX

注意：搞清楚彼此之间的依赖关系（即pytorch依赖CUDA，cuDNN依赖CUDA，CUDA依赖显卡驱动，显卡驱动依赖显卡），才能一次性成功装好合适的版本；

分析：你要安装pytorch框架，进入pytorch官网可见依赖CUDA版本有CUDA10.2和CUDA11.1，那么你就需要安装这二者其一，假设安装CUDA11.1版本，那么我们就要看与CUDA11.1对应的显卡驱动最小的版本是多少，假设为>=450.x，那么你需要安装的驱动就要在450版本以上，而你的显卡也就要支持所选择的版本驱动才行；由此，依次确定安装好显卡驱动、CUDA以及与CUDA相适合的版本cuDNN，再安装pyTorch;（1.根据不同的环境做相应的版本调整；2.考虑版本的兼容性）；

我的选择：pyTorch1.9.0+CUDA11.1+cuDNN8.2.1+驱动470.57.02+GTX-1060