吃了亏还皮

显卡、CUDA、TensorFlow、PyTorch这四个组件想要能够稳定的运行在一个电脑上有点不容易，由于其多样性的版本经常出现各种问题，下面讨论下如何让四个组件协调工作，只考虑Windows系统下。显卡显卡是个人计算机最基本组成部分之一，用途是将计算机系统所需要的显示信息进行转换驱动显示器。显示芯片是显卡的主要处理单元，因此又称为图形处理器(Graphic Processing Uni...

出现问题的时候，我CUDA版本为9.0，升级到10.0就可以了，下载CUDA10.0安装CUDA 10.0地址别忘记安装对应的cuDNN~如果不想安装新的版本CUDA，可以在程序里把 cuDNN禁用即可，在main函数开始加这么一段代码torch.backends.cudnn.enabled = False 程序就可以正常运行了，不过就没有了cuDNN的加速，速度会稍微慢一...