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
相關文章
相關標籤/搜索