当前位置:网站首页>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 不报错即可
边栏推荐
猜你喜欢
随机推荐
APISIX Ingress v1.5-rc1 released
leetcode 5 最长回文子串std::pair 和 make_pair运用
李沐老师 PyTorch版——线性回归 + softmax回归的简洁实现(3)
摔倒检测综述
Introduction to uvm
node模块
手写哈希表
分布式系统API网关原理及选型
Kingbase ES创建过程,报错:PL/SQL block not end correctly
【RPC】Mercury RPC
在AWS CodeBuild中更新Lambda导致进程被占用(status error 255)的解决方案
Horizontal version of the generated image uniapp H5 signature
word写论文中的一些方法
string hash hash value
nacos的配置中心
学习残差神经网络(ResNet)
基于Mysql的智慧医疗管理系统
改变this指向
线上接口流量突增,快要扛不住了
人机对话中得意图识别(关键词提取,svm工具)









