1. 首页
  2. 人工智能
  3. 深度学习
  4. CUDA10.1向下兼容10.0完美运行官方TensorFlow、Pytorch

CUDA10.1向下兼容10.0完美运行官方TensorFlow、Pytorch

上传者: 2020-05-18 07:52:48上传 CMD文件 310B 热度 89次
现在官网提供Tensorflow_gpu、Pytorch都只支持CUDA10.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。
下载地址
用户评论