现在官网提供Tensorflow_gpu、Pytorch都只支持CUDA 10.0,但NVIDIA官网不再更新10.0,很多新的显卡或系统无法安装CUDA10.0。
怎么办?其实可以很方便的使得10.1向下兼容10.0,下载附件的批处理文件并运行,就可以了。
亲测最新CUDA10.1可完美运行TensorFlow13.1、TensorFlow14.0、Pytorch1.2。
本脚本仅支持Windows平台,Linux可以把copy改为link,通样完美运行TensorFlow13.1、Pytorch1.2。