您的当前位置:首页2018-12-25 | 装环境

2018-12-25 | 装环境

2024-12-13 来源:哗拓教育

cuda & cudnn

CUDA is a parallel computing platform and programming model invented by NVIDIA. It enables dramatic increases in computing performance by harnessing the power of the graphics processing unit (GPU).

The NVIDIA CUDA® Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, normalization, and activation layers.


查看cuda 和 cudnn版本

cat /usr/local/cuda/version.txt

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2


theano config文件优先级

The order of precedence is:

- an assignment to theano.config.<property>

- an assignment in THEANO_FLAGS(只读)

- an assignment in the .theanorc file (or the file indicated in THEANORC) (改这个)


查看THEANO_FLAGS

python -c'import theano; print(theano.config)'|less

dnn.include_path (<class 'str'>)

    Doc:  Location of the cudnn header (defaults to the cuda root)

    Value:  /usr/local/cuda/include

dnn.library_path (<class 'str'>)

    Doc:  Location of the cudnn header (defaults to the cuda root)

    Value:  /usr/local/cuda/lib64

dnn.enabled (('auto', 'True', 'False')) Doc: 'auto', use cuDNN if available, but silently fall back to not using it if not present. If True and cuDNN can not be used, raise an error. If False, disable cudnn Value: False

dnn的路径都有,不用自己加,试试自己改enable flag->报错变成了:

UserWarning: Your cuDNN version is more recent than Theano. If you encounter problems, try updating Theano or downgrading cuDNN to version 5.1.

  warnings.warn("Your cuDNN version is more recent than "

Using cuDNN version 6021 on context None

Mapped name None to device cuda: GeForce GTX 1070 (0000:01:00.0)


安装新版本的theano和1.1.0版本的keras

不知道怎么解决这个cudnn问题…不如避开?


装完mkl-service之后又不能用opencv了,烦。


在anaconda cloud里面找不到3.0.0版本的opencv,装了3.1.1版本的。

希望可以用。

显示全文