在Windows 10上使用rtx 2070 gpu的Cuda 10.1在导入tensorflow时不起作用

时间:2019-05-06 20:14:01

标签: python tensorflow jupyter-notebook

嗨,我正尝试在CUDA 10.1中使用tensorflow

我已经为Cuda 10.1安装了cuda10.1和CuDNN 7.5

启动此命令“ nvcc -V”以检查我的安装时,出现此Windows提示符

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:26_Pacific_Standard_Time_2019
Cuda compilation tools, release 10.1, V10.1.105

但是当我在jupyter笔记本上导入tensorflow时,出现此错误

import tensorflow as tf;

错误:

ImportError: Could not find 'cudart64_100.dll'. TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. Download and install CUDA 10.0 from this URL: https://developer.nvidia.com/cuda-90-download-archive

我该怎么做才能使其正常工作? 。预先感谢您的帮助

4 个答案:

答案 0 :(得分:0)

要修复此问题,您需要安装Visual Studio 2017,CUDA 10才能正常工作

答案 1 :(得分:0)

我刚刚使用最新版本的Visual Studio重新安装了tensorflow和cuda 10.0。从内存中,它会警告您并非所有功能都可能正常工作(可能是nvidia NSight)。它不会阻止您继续安装。

答案 2 :(得分:0)

1)卸载CUDA 10.1并安装CUDA 10.0
2)转到cuDNN存档https://developer.nvidia.com/rdp/cudnn-archive
3)单击下载cuDNN v7.6.1(2019年6月24日),适用于CUDA 10.0
       (您需要安装CUDA 10.0。不是10.1。如果安装了错误的版本,请卸载         并安装与tensorflow-gpu兼容的10)
4)单击您的操作系统的链接。
5)解压缩。它应该解压缩到名为CUDA的文件夹。
6)进入CUDA文件夹并复制内容
7)打开已安装的CUDA 10位置。对于Windows 10,它是“ C:\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v10.0”
8)将剪贴板中的内容粘贴到文件夹中。
9)喝咖啡。你完成了!

答案 3 :(得分:0)

如果您使用的CUDA高于10.0,则可以

  • 下载 cudart64_100.dll 并将其手动复制到您的 CUDA bin 文件夹

(类似于C:\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v10.1 \ bin的路径)

请找到以下链接以下载包含该压缩文件的zip文件(摘自CUDA 10.0)

https://www.dropbox.com/s/mz7ewnoc6pfxqql/cudart64_100_dll_.zip?dl=0

完成此操作后,您可能会遇到另一个错误,提示未找到 cublas64_100.dll

  • 从同一文件夹中复制您的 cublas64_10.dll 重命名 cublas64_100.dll