ubuntu系统,安装tensorflow-gpu 解决pip过慢、卡死情况

在这里假设您已经在ubuntu系统下安装好了NVIDAI、CUDA、cudnn等前提条件。 验证是否成功的代码: 检查驱动的命令 nvidia-smi 验证CUDA的命令 nvcc -V 查看cudnn的命令 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 之后可以通过下面一条命令可以快速安装tensorflow: sudo
相关文章
相关标签/搜索