• 售前

  • 售后

热门帖子
入门百科

Win10下安装CUDA11.0+CUDNN8.0+tensorflow-gpu2.4.1+pytorch1.7.0+paddlepa

[复制链接]
塞上云烟辣 显示全部楼层 发表于 2021-10-26 13:03:54 |阅读模式 打印 上一主题 下一主题
目次


  • 下载地点
  • 安装CUDA
  • 设置环境变量
  • 安装CUDNN
  • 验证CUDA是否安装乐成
  • 安装tesorflow-gpu2.4.1
  • 安装pytorch-gpu1.7.0
  • 安装paddlepaddle-gpu2.0.0

下载地点

官方下载:CUDA和CUDNN.






安装CUDA

安装之前,建议关掉360安全卫士



双击cuda_11.0.3_451.82_win10.exe文件



根据自己需要更改安装路径














将Visual Studio Integration的勾去掉

















设置环境变量



  1. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin;
  2. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\libnvvp;
复制代码






安装CUDNN


将下载的CUDNN解压缩,如下图。


将将CUDNN文件夹里面的bin、include、lib文件直接复制到CUDA的安装目次,如下图为CUDA的安装位置,粘贴过来直接覆盖即可。
  1. # CUDA的安装目录
  2. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
复制代码







期待复制完成,即可!


验证CUDA是否安装乐成

打开cmd,输入如下下令,即可!
  1. nvcc -V
复制代码


安装tesorflow-gpu2.4.1

检察对应版本
https://tensorflow.google.cn/install/source_windows#gpu

  1. pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.4.1
复制代码
测试代码
  1. import tensorflow as tf
  2. import os
  3. os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
  4. print(tf.__version__)
  5. a = tf.constant(1.)
  6. b = tf.constant(2.)
  7. print(a+b)
  8. print('GPU:', tf.test.is_gpu_available())
复制代码

安装pytorch-gpu1.7.0

检察对应版本
https://pytorch.org/get-started/previous-versions/

  1. pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html
复制代码

测试代码
  1. import torch
  2. print(torch.__version__)
  3. print(torch.cuda.is_available())
复制代码


安装paddlepaddle-gpu2.0.0

检察对应版本
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/2.0/install/pip/windows-pip.html

  1. python -m pip install paddlepaddle-gpu==2.0.0.post110 -f https://paddlepaddle.org.cn/whl/stable.html
复制代码



测试代码
  1. import paddle
  2. paddle.utils.run_check()
复制代码

到此这篇关于Win10下安装CUDA11.0+CUDNN8.0+tensorflow-gpu2.4.1+pytorch1.7.0+paddlepaddle-gpu2.0.0的文章就先容到这了,更多干系tensorflow pytorch CUDA 安装内容请搜刮草根技术分享以前的文章或继承欣赏下面的干系文章希望各人以后多多支持草根技术分享!

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x

帖子地址: 

回复

使用道具 举报

分享
推广
火星云矿 | 预约S19Pro,享500抵1000!
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

草根技术分享(草根吧)是全球知名中文IT技术交流平台,创建于2021年,包含原创博客、精品问答、职业培训、技术社区、资源下载等产品服务,提供原创、优质、完整内容的专业IT技术开发社区。
  • 官方手机版

  • 微信公众号

  • 商务合作