栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

配置Cuda步骤

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

配置Cuda步骤

配置Cuda步骤
  1. 型号、版本(RTX30xx的显卡还不支持Tensorflow1.X)
  1. 显卡:RTX3080 12G
  2. PYTHON:3.8.0
  3. CUDA:11.1.1
  4. Cudnn:8.0.5.39
  5. Tensorflow-1.15.4:

①Python3.8:https://github.com/fo40225/tensorflow-windows-wheel/issues/167

tensorflow-windows-wheel/1.15.4+nv20.12 at master · fo40225/tensorflow-windows-wheel · GitHub  下载对应的GPU压缩包并解压,将解压后的whl文件直接用pip安装即可。

②Python3.7:Cuda版本应该要更改为以下的

知乎:win10,RTX 3070,可以安装cuda10.0和tensorflow-gpu1.14.0吗? - 知乎

链接:百度网盘 请输入提取码

提取码:dg4q

  1. 安装Cuda、Cudnn:
  1. 下载Cuda:CUDA Toolkit Archive | NVIDIA Developer

①安装(自定义):

展开Driver components,查看“Driver”的版本,如果当前版本大于Cuda所需版本,则不勾选该“Driver”,反之亦然。

2、(需登录)下载对应的Cudnn:cuDNN Archive | NVIDIA Developer

3、拷贝:将Cudnn解压后的 bin、include、lib三个文件夹直接拷贝到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv11.1 下,会自动合并

4、检测是否安装成功:cmd输入nvcc -V

5、tensorflow下载:Links for tensorflow-gpu

6、tensorflow,CUDA,cudnn对应版本号查看:Build from source on Windows  |  TensorFlow

7、torch-gpu下载:https://download.pytorch.org/whl/torch_stable.html

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/997866.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号