首先查看驱动:nvidia-smi
NVIDIA-SMI 436.50
查看对应版本:https://tensorflow.google.cn/install/source_windows#gpu
cuda10.1地址:https://developer.nvidia.com/cuda-toolkit-archive
cndnn地址:https://developer.nvidia.com/rdp/cudnn-archive
在这之前已经安装了cuda8.0,所以现在有两个版本的cuda,分别是cuda10.1和cuda8.0。
那怎么切换使用呢?????
方法一:
1.哪个版本不用时,就把那个环境变量中的path路径改为非实际路径,比如v8.0改为v8.0.111
2.需要用的版本,就在环境变量中将CUDA_PATH,NVCUDASAMPLES_ROOT改成对应的路径
方法二:
1.在需要使用的工具中,创建不同的虚拟环境。比如使用tensorflow框架过程中,那就创建两个不同的虚拟环境,分别安装两个版本的tensorflow,tensorflow会根据自身版本的需求找到对应的cuda版本。在需要使用哪个版本时,激活那个虚拟环境就行。
测试是否安装成功
用管理员身份打开命令行,并转到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite
打开 bandwidthTest.exe
出现 Result = PASS
打开 deviceQuery.exe
出现 Result = PASS
最后都显示 Result = PASS 则说明安装成功
cuda与tensorflow的对应版本:
可在此网址查看https://www.tensorflow.org/install/source#common_installation_problems