Linux使用conda配置兼容TensorFlow和PyTorch的env环境
目录
3、使用conda安装TensorFlow2.2.0,先安装tf !
Tensorflow和pytorch都是机器学习领域的常用框架,导入import时对python的版本和编译环境有要求,如果要想同时使用Tensorflow和pytorch,需将其安装在同一个环境下。自己在尝试很多次之后配置了一套兼容环境,在本环境下,使用TensorFlow的模型和pytorch的模型可以部署在统一系统架构上,直接上步骤:
1、新建虚拟环境:
conda create --name [环境名称] python=3.6
输入y 确认安装:
2、进入环境,检查python版本
-
source activate newEnv
-
-
python --version / -V
3、使用conda安装TensorFlow2.2.0,先安装tf !
conda install tensorflow==2.2.0
检查TensorFlow安装是否成功:
-
>>python
-
-
>>import tensorflow as tf
-
-
>>tf.__version__
4、安装torch1.10版本(确保tf已经没有问题了)
不使用conda!使用pip
-
-
-
pip install torch
检查pytorch:
-
>>python
-
-
>>import torch
-
-
>>torch.__version__
5、检查TensorFlow和pytorch同时导入项目:
vim test.py
-
import tensorflow as tf
-
import torch
-
-
print(tf.__version__)
-
print(torch.__version__)
-
保存退出,执行test.py:
TensorFlow和pytorch同时导入在了一个项目里。
测试不易,如果帮助到了你,点个赞再走吧
这篇好文章是转载于:学新通技术网
- 版权申明: 本站部分内容来自互联网,仅供学习及演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,请提供相关证据及您的身份证明,我们将在收到邮件后48小时内删除。
- 本站站名: 学新通技术网
- 本文地址: /boutique/detail/tanhggabif
系列文章
更多
同类精品
更多
-
photoshop保存的图片太大微信发不了怎么办
PHP中文网 06-15 -
《学习通》视频自动暂停处理方法
HelloWorld317 07-05 -
Android 11 保存文件到外部存储,并分享文件
Luke 10-12 -
word里面弄一个表格后上面的标题会跑到下面怎么办
PHP中文网 06-20 -
photoshop扩展功能面板显示灰色怎么办
PHP中文网 06-14 -
微信公众号没有声音提示怎么办
PHP中文网 03-31 -
excel下划线不显示怎么办
PHP中文网 06-23 -
excel打印预览压线压字怎么办
PHP中文网 06-22 -
TikTok加速器哪个好免费的TK加速器推荐
TK小达人 10-01 -
怎样阻止微信小程序自动打开
PHP中文网 06-13