当前位置:   article > 正文

Ubuntu22.04.3安装pytorch12.1和tensor flow2.15.0_tensorflow 2.15安装

tensorflow 2.15安装

一、安装pytorch12.1

1、安装

(1)打开浏览器,进入pytorch官网选择合适的版本,复制最下面的代码。

9a984583210a4012ad280e19032f8790.png

(2)打开终端,创建虚拟环境

 conda create -n torch12.1cu12.2

101b9f61c73440689b14450e9a266812.png 

(3)激活并进入虚拟环境

conda activate torch12.1cu12.2

(4)将刚才的代码粘贴进行安装

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch-c nvidia(以官网为准)

d02b9bc01fb94af2a718922e071e8385.png

 2、检验

(1)在终端中打开python

(2)调用torch进行检验

(3)如果有结果则下载成功

在终端输入:

python

 import torch

 print(torch.__version__)

print(torch.version.cuda)

print(torch.backends.cudnn.version())

adb13859aeb5477e9ddb109d8f7ea04e.png

二、下载tensorflow2.15.0

1、下载

(1)打开终端,创建环境 

conda create -n tf1 python=3.11

(2)激活并进入环境

conda activate tf1

(3)镜像源,打开浏览器搜索pip清华镜像在下图所示页面复制黑框中的代码,粘贴到终端后回车(可以不用镜像,但下载会慢)

f101a5dfd1df4854ada6b2ced15ee7b4.png

750b4a80aef74e7aa0fdba3498a0eb3a.png 

(4)安装:在终端输入下面代码,后回车

pip install tensorflow ==2.15.0

ea87feefde424ca28a3ea8d583f66eae.png

(5)安装完毕后输入

pip install opencv- python

2、检验

(1)在终端中打开python

(2)调用tensorflow进行检验

(3)如果有结果则下载成功

在终端输入;

python

import tensorflow as tf

89d7aaff5325411a81282e24ba36a02b.png

 

 

本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号