跳坑成功!(若是看到这个的东西,安装出问题,请给我留言或者私信,我来帮你跳坑)
======== 以下为第二版=======
cudu 驱动下载:https://www.geforce.com/drivers
cudnn下载 https://developer.nvidia.com/cuda-downloads ,这个是最新版本的,如果要下载某些版本的可点击:
查看GPU对应的cuda 请看: https://developer.nvidia.com/cuda-gpus
tf - >cudnn -> cuda 过程是上面的。cafe,mxnet,tf都是基于cuda编程的。
cudnn下载:https://developer.nvidia.com/cudnn
为如下:
第一步:解压
第二步:修改注册表
第三步:重启电脑。
第四步:检测tensorflow是否使用GPU
第五步:检测pytorch是否使用GPU
======== 以下为第一版=======
环境:windows,1050ti ,电脑显卡驱动已经装好。
- 先说个很重要的话,你先去看下你的显卡是否可以用cuda加速,去官网看下就可以了,可以加速都给出来了。比如我的1050(1050ti 包含在这)。
- 然后看下你要装什么版本的tensorflow,比如我要装1.8的,一查可以用cuda9.0,cudnn7.0(cudnn9.0)都合适,主要是看tensorflow版本要和什么cuda和cudnn版本配合。
想了解如何选择和正确安装跟Tensorflow版本对应的CUDA和cuDNN版本请看:
https://www.tensorflow.org/install/source_windows
参考:https://blog.csdn.net/XCCCCZ/article/details/80385448