安装gpu版的tensorflow相对CPU版的来说复杂的多,因为要安装英伟达专门针对gpu计算的平台cuda和加速库cudnn,这两个安装比较麻烦。
1. 双显卡的同学先把security boot给disable掉
2.禁用nouveau驱动(十分重要)
3.根据自己的情况安装对应的英伟达驱动
4.把gcc版本降到5.0及以下
5.安装cuda和cudnn
6.安装tensorflow
安装gpu版的tensorflow相对CPU版的来说复杂的多,因为要安装英伟达专门针对gpu计算的平台cuda和加速库cudnn,这两个安装比较麻烦。
1. 双显卡的同学先把security boot给disable掉
2.禁用nouveau驱动(十分重要)
3.根据自己的情况安装对应的英伟达驱动
4.把gcc版本降到5.0及以下
5.安装cuda和cudnn
6.安装tensorflow