首页
学习
活动
专区
工具
TVP
发布
您找到你想要的搜索结果了吗?
是的
没有找到

tensorflow安装GPU版本

tensorflow安装GPU版本主要要点 1.先通过该网站查看tensorflow和cuda和cudnn版本以及visual studio(MSVC)的对应关系。...cuda的版本(这里指的是你的电脑支持的最高版本的cuda driver版本,并不是推荐你安装这个,仅参考) 选择cuda的版本比较麻烦,需要兼顾tensorflow,显卡驱动版本,电脑版本。...2.在这个网站上根据自己需要下载对应版本的cuda toolkit,如果网络很好可以选择下载较小的network包,一般建议安装local本地包。...https://developer.nvidia.com/cuda-toolkit-archive 安装时可以选择精简安装,也可以自定义安装,可选择想安装的插件,一般可以默认全选。...__version__) print("use GPU", gpu_ok) 我的环境以及最终安装完成的版本(rtx2060驱动版本451.67,cuda10.0.130,cudnn7.6.2.24,tensorflow-gpu1.15.4

48610
领券