Anaconda下的tensorflow-gpu2.6.0安装使用

引言:本次安装全部是在window下安装的虚拟环境,电脑显卡为1650ti,给相同配置的朋友一点参考,也给自己记录一下整体过程。

1.创建虚拟环境

  1. 在已经安装anaconda的情况下,搜索点开anaconda prompt
  2. 创建python版本为3.9.0的虚拟环境用于搭建tensorflow框架 ,命令格式conda create -n name python=x.x.x,我的命令conda create -n tensorflow2.6.0 python=3.9.0 ,这里安装其他版本的python也可以只要和tensorflow安装版本兼容即可。
  3. activate tensorflow2.6.0 激活,要进入到tensorflow里去下载包。

2.虚拟环境配置

  1. 接下来需要根据你自己电脑配置选择合适的包安装。

    tensorflow官方安装问题该网址可以查看什么cuda对应什么cudnn对应什么版本的tensorflow。

    怎么选择cuda版本可以参考《主博客 Win10安装Anaconda3、Python、TensorFlow(GPU和CPU版本)、Keras(特别是版本选择)》这篇博客,里面教你如何通过指数值选择合适自己的cuda。

在这里插入图片描述

### 安装TensorFlow-GPU和PyTorch在Anaconda环境中的方法 #### TensorFlow-GPU安装 为了确保兼容性和性能优化,可以使用Python 3.7版本来安装TensorFlow-GPU 2.6.0。以下是具体的配置命令: ```bash # 设置镜像源以加速下载过程 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ # 使用Mamba加速依赖解析并安装指定版本的tensorflow-gpu micromamba install tensorflow-gpu==2.6.0[^1] ``` 对于CPU-only版本,则可以直接执行如下指令: ```bash micromamba install tensorflow==2.6.0 ``` 验证GPU支持情况可以通过下面这段简单的测试脚本完成: ```python import tensorflow as tf print(tf.__version__) print('GPU Available:', tf.config.list_physical_devices('GPU')) ``` 如果一切正常,应该能够看到类似于`[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]`这样的输出。 #### PyTorch及其CUDA扩展库的安装 考虑到不同版本之间的匹配问题,建议先确认目标机器上已有的CUDA版本号再做决定。假设当前环境中存在CUDA 10.2的话,那么就可以按照以下方式操作: ```bash # 同样推荐采用清华镜像站提高速度 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ # 假设需要安装适用于CUDA 10.2的特定版次 micromamba install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch[^4] ``` 上述过程中涉及到了多个软件包的选择与组合,其中`torchaudio`是一个可选组件,主要用于处理音频数据;而`cudatoolkit`则是用来指明所使用的CUDA工具链的具体版本。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值