当前位置：网站首页>CUDA10安装支持gpu的tensorflow版本

CUDA10安装支持gpu的tensorflow版本

2022-08-08 06:22:00 【feifeiyechuan】

由于英伟达官方宣称CUDA10有着巨大的性能提升，我便迫不及待的想尝试用它来训练深度学习模型，然而现在Tensorflow的最新官方GPU版本1.12并不支持在CUDA10下的模型训练，因此便翻墙到国外论坛找材料，最终找到了可以完美适配CUDA10的tensorflow方案。具体的环境如下：


安装步骤：

首先将相关的轮子依赖下载到本地系统中，（https://download.csdn.net/download/weixin_38538282/10816219或者翻墙下载：https://1drv.ms/u/s!AspzPyzQ0HQR7VNCicPlnCV2abKJ）我选择的是Pycharm工程venv目录下的Scripts，然后再进入到该目录下，输入pip3 install tensorflow-1.12.0rc0-cp36-cp36m-win_amd64.whl 命令，即可成功安装CUDA10下支持的tensorflowGPU版本。

验证：进入python

输入：pip install tensorflow 不报错即可


 

原网站

版权声明
本文为[feifeiyechuan]所创，转载请带上原文链接，感谢
https://sixiangdefairy.blog.csdn.net/article/details/94480078

边栏推荐

猜你喜欢

随机推荐