https://blog.csdn.net/sinat_28442665/article/details/107384944
是按照这个安装的,但是当我重新用xshell连接服务器的时候使用命令nvcc -V
就找不到cuda信息。
解决办法:需要在添加环境变量时所处的路径下,重置一下配置文件。
重置命令source .bashrc
,再输入nvcc -V
就可以看到了。
1.查看Tensorflow版本:
import tensorflow as tf
tf.__version__
- 查看CUDA的版本 :
cat /usr/local/cuda/version.txt
- 查看CUDNN的版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
网友评论