Win10+TensorFlow(GPU)+MX450
本文参考了以下两篇文章:
-
首先看官网给出的软件要求
-
在桌面右击鼠标打开NVIDIA控制面板,点击左下角系统信息
-
查看CUDA版本
-
在网站CUDA Toolkit下载对应版本(我这里是11.0)
-
下载,比较慢,可以用加速插件
-
双击安装程序,如果没有安装VS,就取消VS勾选,然后一直下一步安装
-
打开终端输入nvcc -V,输出以下字符安装成功
-
在网站cuDNN里下载cuDNN,要注册账号并同意条款才能下载,选择对应版本及操作系统
-
解压压缩包
-
将解压的包中的文件复制过去,如图所示
-
在终端输入nvidia-smi查看是否安装成功,安装成功如图所示
-
在终端导入tensorflow,并输入
tf.config.list_physical_devices(‘GPU’)
如图所示则安装成功