最新tensorflow与pytorch环境搭建的实现步骤

 更新时间:2024年04月25日 08:35:07   作者:DreamNotOver  
深度学习相关的热门框架主要为Tensorflow和Pytorch,本文主要介绍了搭建最新tensorflow与pytorch环境,具有一定的参考价值,感兴趣的可以了解一下

1、安装 Anaconda:

如果您尚未安装 Anaconda,首先访问 https://www.anaconda.com/products/distribution/ 下载适用于您操作系统的最新版本。按照官方指南完成安装过程。

2、设置 Conda 源

方法一:命令行配置

临时使用: 如果您只想临时为一次命令使用清华源,可以在安装或更新包时直接指定源:

conda install <package-name> -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/<channel-name>

请将 <package-name> 替换为您要安装的包名,<channel-name> 替换为具体的频道名,如 conda-forge、pytorch 等。例如:

conda install numpy -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge

永久更改: 若要永久将 Conda 的默认源设置为清华源,执行以下命令:

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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --set show_channel_urls yes

方法二:编辑用户目录的.condarc 文件

打开配置文件: 使用文本编辑器打开您的 Conda 配置文件:

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - defaults
show_channel_urls: true
ssl_verify: false #解决https error问题

3、设置 pip源

方法一:临时使用

在安装单个包时临时使用清华源:

pip install --index-url https://pypi.tuna.tsinghua.edu.cn/simple <package-name>

将 <package-name> 替换为您要安装的包名。

方法二:永久更改

你可以通过使用 pip config 命令来生成

pip config set global.index-url Simple Index

这个命令将在你的用户主目录下的 .pip 文件夹中创建或更新 pip.conf 文件,然后将清华源添加到其中。

conda config --set ssl_verify false

 此命令禁用 SSL 证书验证。注意,这样做会降低安全性,因为您将不再验证下载包的完整性。完成操作后记得重新启用证书验证。

源地址变动:确认源地址:访问清华大学开源软件镜像站(如 Index of /anaconda/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror ),查看最新的 Conda 镜像源地址。有时,源地址可能会发生变化,您需要更新配置中使用的 URL。

官方建议:按照清华大学提供的官方指南(如 anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror )来配置 Conda 源,确保使用的是官方推荐的最新地址和方法。

4、安装tensorflow

4.1创建新的虚拟环境:打开 Anaconda Prompt(Windows)或终端(Linux/macOS),运行以下命令创建一个新的 Conda 虚拟环境,这里假设环境名为 tf-env。您可以根据需要替换为其他名称: conda create -n tf-env =3.9

4.2 创建环境后,激活它:conda activate tf-env

4.3在激活的环境中,使用以下命令安装最新版本的 TensorFlow

pip install tensorflow  或conda install tensorflow

注意:从 TensorFlow 2.7 开始,官方推荐使用 tensorflow 包,它会自动选择 CPU 或 GPU 版本(如果适用)。

4.4验证安装:

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

或 安装完成后,打开  解释器或 Jupyter Notebook,输入以下代码

import tensorflow as tf

print(tf.__version__)

5、安装 JupyterLab

JupyterLab 默认已经包含了基于 jedi 或 ipywidgets 等库的代码补全功能。您只需启动 JupyterLab,然后在编写代码时按 Tab 键即可触发自动补全。但您可能希望进一步增强其功能,例如使用更先进的语言服务器协议 (Language Server Protocol, LSP) 支持。JupyterLab LSP 提供了一个框架,用于集成各种语言的 LSP 服务,以获得更强大的代码补全、语法检查、跳转到定义等功能。

用以下命令通过 pip 安装 JupyterLab jupyterlab-lsp:

pip install jupyterlab  jupyterlab-lsp

修改配置文件

生成配置文件(如果尚未存在):jupyter lab --generate-config

这将在您的用户家目录下创建一个名为 jupyter_lab_config.py 的文件,通常位于 ~/.jupyter/jupyter_lab_config.py(具体路径可能因操作系统而异)。

在配置文件中找到如下一行(可能被注释掉):

# c.ServerApp.root_dir = ''

取消注释(去掉 #),并将其修改为您希望 JupyterLab 启动时默认打开的路径:

c.ServerApp.root_dir = '/path/to/your/workdir'

请确保将 /path/to/your/workdir 替换成实际的完整路径。

保存并关闭文件:

保存对配置文件所做的修改,然后关闭文本编辑器。

启动命令:  jupter  lab

6、安装其他包:

conda  install  matplotlib  scikit-learn  jieba  pandas  seaborn

7、搭建pytorch环境

conda create -n torch-env python=3.9

此命令将创建一个使用  3.9 的新环境。

创建环境后,激活它:

conda activate torch-env

转到 PyTorch 官方网站 https://pytorch.org/get-started/locally/ ,根据您的操作系统、 版本、CUDA 版本(如果有 NVIDIA GPU)以及是否需要额外的库

例如,如果您有一个支持 CUDA 12.x 的 NVIDIA GPU,并希望同时安装 torchvision,复制提供的 conda install 命令(通常类似于以下格式)并粘贴到终端中执行:

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

如果您没有 NVIDIA GPU 或不打算使用 GPU 加速,选择对应的 CPU 版本命令,例如

conda install pytorch torchvision torchaudio cpuonly -c pytorch

安装完成后,打开  解释器或 Jupyter lab,在其中输入以下代码验证 PyTorch 是否安装成功并查看版本信息:

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果安装了 GPU 版本,此行将显示是否正确识别到 CUDA

如果没有报错且输出了 PyTorch 的版本号,并且(对于 GPU 版本)torch.cuda.is_available() 返回 True,说明安装成功。

至此,您已使用 Anaconda 成功搭建了包含 PyTorch 的虚拟环境。在需要使用该环境时,只需通过 conda activate pytorch-env 命令激活即可。要退出当前环境,运行 conda deactivate。

到此这篇关于最新tensorflow 与pytorch环境搭建的实现步骤的文章就介绍到这了,更多相关tensorflow 与pytorch环境搭建 内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • pytorch中torch.topk()函数的快速理解

    pytorch中torch.topk()函数的快速理解

    我们在做分类算法时,时常见到@acc1和@acc5的情况,@acc1比较容易实现,但是一直苦于@acc5算法的实现,在此为大家提供一种@topk的实现方法,这篇文章主要给大家介绍了关于pytorch中torch.topk()函数的快速理解,需要的朋友可以参考下
    2022-02-02
  • python3 os进行嵌套操作的实例讲解

    python3 os进行嵌套操作的实例讲解

    在本篇文章里小编给大家整理了关于python3 os进行嵌套操作的实例内容,有兴趣的朋友们可以学习下。
    2020-11-11
  • Python类继承及super()函数使用说明

    Python类继承及super()函数使用说明

    这篇文章主要介绍了Python类继承及super()函数使用说明,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2022-11-11
  • 基于python实现的抓取腾讯视频所有电影的爬虫

    基于python实现的抓取腾讯视频所有电影的爬虫

    这篇文章主要介绍了用python实现的抓取腾讯视频所有电影的爬虫,这个程序使用芒果存, 所以大家需要下载使用mongodb才可以
    2016-04-04
  • Python正则表达式的另类解答

    Python正则表达式的另类解答

    这篇文章主要为大家详细介绍了Python正则表达式,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下,希望能够给你带来帮助
    2022-02-02
  • Python 一篇文章看懂Python集合与字典数据类型

    Python 一篇文章看懂Python集合与字典数据类型

    集合并不是一种数据处理类型,而是一种中间类型。集合(set)是一个无序、不重复的元素序列,经常被用来处理两个列表进行交并差的处理性。本文将详细讲解集合的一些常用方法,感兴趣的可以了解一下
    2022-03-03
  • python中扫描条形码和二维码的实现代码

    python中扫描条形码和二维码的实现代码

    pyzbar模块是Python一个开源库用于扫描和识别二维码信息。这篇文章主要介绍了python中扫描条形码和二维码的示例代码,需要的朋友可以参考下
    2021-10-10
  • Python容器类型使用新手必读

    Python容器类型使用新手必读

    本文将介绍一些Python容器类型的使用小技巧,以便更高效地处理数据和提升编程技能,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-11-11
  • python中as用法实例分析

    python中as用法实例分析

    这篇文章主要介绍了python中as用法,实例分析了as的功能及相关使用技巧,非常具有实用价值,需要的朋友可以参考下
    2015-04-04
  • python 操作excel表格的方法

    python 操作excel表格的方法

    这篇文章主要介绍了python 操作excel表格的方法,帮助大家更好的理解和使用python,感兴趣的朋友可以了解下
    2020-12-12

最新评论