win10cuda回退/TensorFlow-cpu转gpu

cuda降级回退

	现在官网上的cuda是10.2的,但是现在pytorch支持的cuda是10.1,tensorflow所支持的cuda是10.0。

1.cuda10.2删除
我按照安装时间把NVIDIA软件除NVIDIA PhysX系统软件和图形驱动程序之外的所有都删掉了。有的不知道可以不可以删的,上网也不能确定,也一并删了。
2. 下载安装10.0既可。

TensorFlow-cpu转gpu

 之前我用anaconda下载了TensorFlow-cpu版,现在要改成gpu版本。
  1. 先删除TensorFlow旧版本
pip uninstall protobuf
pip uninstall tensorflow
  1. 下载tensorflow-gpu
pip install tensorflow-gpu==2.0.0

cuDNN 安装

官网下载。
压缩包中bin\lib\include分别对应放到cuda所文件的bin\lib\include中即可

测试tensorflow是否安装成功

tf.test.is_gpu_available()

你可能感兴趣的:(win10cuda回退/TensorFlow-cpu转gpu)