最近在Ubuntu中安裝了TensorFlow的GPU版本,和windows一樣還是先安裝cuda,再安裝cudnn,最後安裝TensorFlow-gpu,這裏同樣特別需要注意版本號對應,另外一些操作都是在終端裏進行的,下面總結一些終端裏常用操作。
0. 查看GPU使用情況
nvidia-smi
watch -n 5 nvidia-smi 每5秒更新一次
1 . 查看CUDA的版本cat /usr/local/cuda/version.txt
或者 nvcc --version
2.查看cudnn的版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
3. 終端裏跳過長文
按q
4. 終端裏複製粘貼
粘貼是ctrl+shift+v
對於Windows用戶來說最好用鼠標複製,因爲在終端了ctrl+c是停止工作,而ctrl+shift+c纔是複製