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。

Anaconda下的tensorflow-gpu2.6.0安装使用
  1. 我的cuda是11.2 ,cudnn是8.1,tensorflow是2.6.0
  2. 在安装包之前可以导入一些镜像源方便快速下载工具包。
  3. 接下来是安装cuda和cudnn工具包,记住在安装tensorflow之前安装!!!
conda install cudatoolkit=11.2
conda install cudnn=8.1
  1. 完成环境安装之后可以安装tensorflow了,pip install tensorflow-gpu==2.6.0 这里使用pip进行安装。
  2. 输入代码测试
python
import tensorflow as tf
tf.__version__
import keras

依然可以参考之前《主博客 Win10安装Anaconda3、Python、TensorFlow(GPU和CPU版本)、Keras(特别是版本选择)》这篇博客

3.pycharm测试

pycharm中file->setting可以打开设置表,右边一个设置按钮,点击可以添加或者显示全部环境,我们添加环境->conda环境->现有环境->…,选择anaconda文件夹->envs文件夹->你的虚拟环境->python.exe,最后全部确定就可以在pycharm上使用该环境了。

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

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

让我们找一个代码来测试它是否工作。

[En]

Let’s find a code to test whether it works.

import tensorflow as tf
print(tf.test.is_gpu_available())
print('tf.version:\n',tf.__version__)

测试结果为true说明tensorflow-gpu可用,2.6.0显示tensorflow版本号

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

4.其他问题

1.首先2.0版本以上的tensorflow里面已经自带keras了,我们可能会在深度学习的过程中遇到tensorflow.python.keras 和keras版本不兼容的问题,这个时候统一使用tensorflow内嵌的keras ,一致使用 from tensorflow.python.keras代替from keras部分。

2.还有可能是导入的库里面没有所需类,这个时候可以去site-packages里面找到具体路径之后再写,比如

from tensorflow.python.keras.optimizer_v2.adam import Adam

5.常用命令

  • conda info -e (查看_所有的虚拟 _环境)
  • activate -name (虚拟 _环境_名字)(进入到该虚拟 _环境_中)
  • conda deactivate 选择退出环境
  • conda remove -n yourenvname --all 删除环境

镜像源

  1. 显示镜像源 conda config --show channels
  2. 添加新镜像源 conda config --add channels [urls…]
  3. 删除镜像源
    conda config --remove-key channels 删除所有的镜像源,恢复到默认
    conda config --remove channels [urls] 删除指定的镜像源
  4. 设置搜索时显示通道地址
conda config --set show_channels_urls yes
  1. 国内常用源镜像地址:

清华:https://pypi.tuna.tsinghua.edu.cn/simple
阿里云:https://mirrors.aliyun.com/pypi/simple/
中国科技大学: https://pypi.mirrors.ustc.edu.cn/simple/
华中理工大学:https://pypi.hustunique.com/
山东理工大学:https://pypi.sdutlinux.org/
豆瓣:https://pypi.douban.com/simple/

  1. conda 安装指定的源
conda install -c urls lib_name
  1. 我使用的镜像源为:
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

(该镜像网址很快很好用>_

Original: https://blog.csdn.net/qq_45156353/article/details/124713124
Author: 硕棋
Title: Anaconda下的tensorflow-gpu2.6.0安装使用

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/497390/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球