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

相关推荐
GuWenyue20 小时前
放弃云端API!一套React+WebGPU本地LLM方案,零数据上传、离线可用
前端·人工智能·前端框架
love530love21 小时前
OpenClaw Windows Companion 桌面客户端 连接 LM Studio 完整配置指南
人工智能·windows·python·openclaw
墨舟的AI笔记21 小时前
从文本提示到可交互道具:AIGC 生成游戏资产的语义对齐与合规校验
人工智能
rain_sxr1 天前
逼近上限就裁剪:大模型 Token 计数与前端上下文预算管理
人工智能
资深数据库专家1 天前
月之暗面Kimi:K3之后,开源怎么走?
人工智能
小林ixn1 天前
告别“屎山”与“幻觉”:3个核心心法,让你的Vibe Coding体验起飞
人工智能·agent
汤姆小白1 天前
06-CLI命令参考
人工智能
颜酱1 天前
04 | 召回前置准备:搭好召回所需的四个数据库
前端·人工智能·后端
A洛1 天前
Codex 实战:一句话完成 Temu 商品图采集与印花抠图自动化
运维·人工智能·自动化·codex