Windows 安装Tensorflow2.1、Pycharm开发环境

文章目录

1、安装anaconda

https://www.anaconda.com/download

打开命令行工具,出现base就表示安装成功了,表示当前的虚拟环境名


2、安装Tensoflow

2.1、创建虚拟环境

在刚才的命令行中进行操作

bash 复制代码
# 创建一个名字为TF2.1的python3.7版本的虚拟环境
conda create -n TF2.1 python=3.7

# 进入虚拟环境(会发现(base)变成了TF2.1表示当然进入了我们创建的虚拟环境)
conda activate TF2.1

2.2、安装Tensorflow依赖

bash 复制代码
# 如果不支持GUP则跳过这两步
# 英伟达SDK=10.1
conda install cudatoolkit=10.1

# 英伟达深度学习软件包7.6
conda install cudnn=7.6

安装tensorflow
pip install tensorflow==2.1

2.3、验证Tensorflow是否成功

查看版本号,如果版本号输出时2.1表示tensorflow安装成功

python 复制代码
python
import tensorflow as tf
tf.__version__

3、配置pycharm环境

打开pycharm创建项目,选择刚才通过anaconda创建的虚拟环境进行开发

创建成功之后,输入以下代码进行验证,第一行时获取版本号,第二行时获取当前是否支持GPU,后面的计算两个变量之和

python 复制代码
import tensorflow as tf


def print_hi(name):
    # Use a breakpoint in the code line below to debug your script.
    print(f'Hi, {name}')  # Press Ctrl+F8 to toggle the breakpoint.


def tensorflow_t():
    tensorflow_version = tf.__version__
    gpu_available = tf.test.is_gpu_available()
    print("tensorflow version :", tensorflow_version, "\tGPU available:", gpu_available)

    a = tf.constant([1.0, 2.0], name="a")
    b = tf.constant([1.0, 2.0], name="b")
    result = tf.add(a, b, name="add")
    print(result)


if __name__ == '__main__':
    tensorflow_t()

4、错误记录

Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found
如果出现cudart64_101.dll找不到去官网进行下载一个放到C:\Windows\System32目录下

官网网址:https://www.dll-files.com/cudart64_101.dll.html

相关推荐
锋行天下4 小时前
langgraph 后端发自定义数据,前端接收方法概括
人工智能·python
泯泷6 小时前
单个 AI 再强,为什么也干不过一支 AI 团队?
人工智能·llm·agent
浪遏6 小时前
认识 wm:从多平台发布到可扩展 CLI 的架构设计
人工智能
深蓝AI6 小时前
英伟达换帅当天,我先跑了一遍 GPU 环境体检:版本矩阵比人事更值得盯
人工智能·gpu
罗西的思考8 小时前
【轻量化复刻大模型 / KIMI K3】AttnRes 模块交互式演示
人工智能·算法·机器学习
stormzhangV8 小时前
涨薪 50% 怎么做到的?
人工智能·面试·aigc
老金带你玩AI9 小时前
如何在5分钟之内创建Grok Bot的超级团队
人工智能
程序员cxuan9 小时前
离谱,AI 大佬顺手搓了个 YouTube ,而我顺手给他搓了个中文插件
人工智能·后端·程序员
zixu_star9 小时前
上下文工程:Prompt、记忆和上下文如何管理?
人工智能·prompt
ZONGYUAN-ROOT9 小时前
【3D仿真数据集】从“能生成”到“能用”:面向工业仿真的3D资产自治自愈流水线全链路落地
人工智能