配置TensorFlow和Pytorch环境(CUDA&cuDNN)

  • 在Windows中安装相应版本的TensorFlow
    • CUDA历史版本CUDA Toolkit Archive | NVIDIA Developer
      • CUDA于显卡驱动版本无关,只要显卡驱动装到最新版本即可
        • 对于CMD中“nvidia-smi”的结果中的CUDA

        • 这里CUDA 12.0的意思是此驱动兼容CUDA12.0以下的所有版本(当前的CUDA只更新到了11.8)
      • CUDA安装
        • 选择自定义

        • 以下四个选项无需安装

          • 因为驱动和PhysX已经最新了而Nsight不怎么用
    • 装完以上CUDA和cuDNN就可以在虚拟环境中运行“conda install tensorflow-gpu==1.13.1”(好像会自动把CUDA和cuDNN打上)
    • 检测TensorFlow是否能使用CUDA进行GPU计算
      • Python控制台下
      • import tensorflow as tf
      • print(tf.test.is_gpu_available())
  • 在Windows中安装相应版本的Pytorch
    • 打开Previous PyTorch Versions | PyTorch找到如果安装某版本Pytorch,查找表首先安装好对于的CUDA


      以上图片就是如果利用CUDA10.1,就先安装好CUDA10.1
    • 安装CUDA同TensorFlow
    • 然后直接复制命令到conda虚拟环境进行安装
    • 检测Pytorch是否能使用CUDA进行GPU计算
      • Python控制台下
      • import torch
      • print(torch.cuda.is_available())
  • 经验
    • 只要前置装了CUDA和cuDNN就可以了,在conda中装TensorFlow和Pytorch会自动下载对应的cudatoolkit版本对接系统的CUDA版本(不知道这个理解对不对)
    • TensorFlow虚拟环境下的condalist

    • Pytorch虚拟环境下的conda list

      • 第二张图突然觉得Pytorch好用了,直接表明了需要装的版本
  • 一些报错
    • conda 安装TensorFlow-Gpu出现权限不够提示加上“--user”,不知道是以下哪种情况导致的,但是可以解决
      • Pycharm未关闭
      • 没有加——user
    • 运行报错“缺少cudart64_110.dll”
      • 意思就是缺少cuda11.0+
    • AttributeError: module 'tensorflow' has no attribute 'variable_scope'
      • 原因就是1.0版本不能用2.0跑
      • 解决方案
        • tensorflow早已进入2.0时代,但是大部分代码还是在1.0时代创造的,修改代码换环境换安装包等等方法令人烦躁,还会引出一堆其他问题,因此给出修复大部分错误的简单解决方法:(未实践证明)
          • 使用tensorflow 2.0以上版本,并且安装tf_slim
          • 将import tensorflow as tf 改成import tensorflow.compat.v1 as tf(不要管ide的红线报错)import tf_slim as slim
          • 添加tf.disable_v2_behavior()到首行即可修复
        • 直接创建虚拟环境配置成老版本的TensorFlow并下载对应的依赖
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值