cuda1.10
今天尝试安装cuda1.10版本。为什么会安装这个版本的cuda,是因为在执行下列语句后会提示缺少cuda :
1 | pip install tensorflow |
提示安装cuda1.10
于是按照tf2和tf1最新版都是同一时刻发布的,于是尝试下cuda10可否支持tf2。经过验证是可行的。
总结
cuda的版本很重要
windows查找对应的cuda版本
今天在网上查到:
在NVIDAI显卡驱动里可以找到相对应的cuda版本为:10.1.131。但是官方没有这个版本的cuda,先测试高版本的cuda_cuda_10.1.168。
并下载对应的vs版本,尝试使用编译的方法安装,安装过程出现意外的BUG,解决BUG后继续安装。
How to install tensoflow-gpu on windows10
Use conda to create a TF2 virtual environment, python version is 3.6, use pip to install tensorflow, and in the import tensorflow according to the error prompt, install the corresponding cuda version.
The following are the instructions used.