虚拟环境中Tensorflow-gpu安装记录

指路链接TensorFlow安装和下载(超详细)_tensorflow下载-优快云博客

Anaconda安装-超详细版(2023)_anaconda安装哪个版本好-优快云博客

Tensorflow-gpu保姆级安装教程(Win11, Anaconda3,Python3.9)-优快云博客

Windows系统上安装TensorFlow(Win10, Anaconda3,Python3.9)

1. 安装Anaconda

Anaconda安装-超详细版(2023)_anaconda安装哪个版本好-优快云博客

2. 创建虚拟环境

打开开始—>点击Anaconda Prompt—>输入

conda create --name tf_gpu pyhton=3.9

3.  激活虚拟环境

conda activate tf_gpu

4.Tensorflow-gpu版本安装的准备工作

Anaconda安装-超详细版(2023)_anaconda安装哪个版本好-优快云博客  前言部分

5. 安装cuda(根据上一步,选择自己需要的cuda版本)

conda install cudatoolkit==11.2

6. 安装cudnn(不写版本会自动选版本与cuda版本相匹配)

conda install cudnn

7. 安装tensorflow-gpu(选择自己需要的版本)

pip install tensorflow-gpu==2.7.0 -i  https://pypi.mirrors.ustc.edu.cn/simple  

8. 验证是否安装成功

python

import tensorflow as tf

tf.__version__

报错解决:

(tensorflow) C:\Users\Rmzh>python
Python 3.9.16 | packaged by conda-forge | (main, Feb  1 2023, 21:28:38) [MSC v.1929 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "D:\WorkSoftware\Install\Anaconda3\envs\tensorflow\lib\site-packages\tensorflow\__init__.py", line 41, in <module>
    from tensorflow.python.tools import module_util as _module_util
  File "D:\WorkSoftware\Install\Anaconda3\env
### 配置 VSCode 使用 Anaconda 创建的 TensorFlow-GPU 环境 #### 1. 安装并配置 Anaconda 和 TensorFlow-GPU 确保已通过 Anaconda 成功创建了一个支持 GPUTensorFlow 环境。以下是具体操作: - **创建虚拟环境** 使用 `conda` 命令创建一个新的 Python 虚拟环境,并指定所需的 Python 版本以及安装 TensorFlow-GPU[^3]: ```bash conda create -n tf-gpu python=3.6 conda activate tf-gpu conda install tensorflow-gpu==1.7 ``` - **验证安装成功** 运行以下代码片段来确认 TensorFlow 已正确安装并且能够检测到 GPU 设备[^2]: ```python import tensorflow as tf print(tf.__version__) sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) ``` #### 2. 在 VSCode 中设置 Python 解释器 为了使 VSCode 使用上述创建的 Conda 环境作为其默认解释器,需完成如下步骤: - 打开 VSCode 并按快捷键 `Ctrl+Shift+P` 或者点击顶部菜单栏中的“视图 -> 命令面板”。 - 输入 `Python: Select Interpreter` 并回车。 - 在弹出的选择列表中找到之前创建好的名为 `tf-gpu` 的 Conda 环境路径(通常类似于 `/path/to/anaconda/envs/tf-gpu/bin/python`),然后选中它。 此过程会自动更新 `.vscode/settings.json` 文件以记录所选用的 Python 解析程序位置[^1]。 #### 3. 配置调试选项 (Optional) 如果计划利用内置调试功能运行项目,则还需要进一步调整 launch configuration 设置: - 在左侧活动栏里选择 “Run and Debug”,再点击齿轮图标生成新的 launch.json 文件; - 修改其中的内容使其指向刚才选定之解析器及其参数, 如下所示: ```json { "name": "Python: Current File", "type": "python", "request": "launch", "program": "${file}", "console": "integratedTerminal" } ``` 至此便完成了全部准备工作,在后续开发过程中可以直接借助于该集成化平台开展工作而无需额外考虑底层依赖关系等问题了。 ```python import tensorflow as tf print("TensorFlow version:", tf.__version__) # Check available devices. from tensorflow.python.client import device_lib local_devices = device_lib.list_local_devices() gpu_names = [x.name for x in local_devices if x.device_type == 'GPU'] if not gpu_names: raise Exception('No GPUs found.') else: print(f'GPUs detected: {", ".join(gpu_names)}') ```
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值