最近在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才是复制