搭建TensorFlow+pytorch环境：jupyternotebook使用

1. tensorflow安装

1.1 安装Anaconda

从官网下载：https://www.anaconda.com/download/

官网下载起来很慢，国内清华镜像网站：https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

1.2 安装Tensorflow

打开安装好的Anaconda中的 Anaconda Prompt
添加镜像

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 --set show_channel_urls yes

添加PyTorch清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud//pytorch/

创建环境

conda create -n ptts python=3.7
查看环境
conda env list

激活

activate ptts

安装

pip install tensorflow

在cmd中依次输入

python
然后键入
import tensorflow as tf
hello = tf.constant(‘Hello tensorfolw’)
sess = tf.Session()
print(sess.run(hello))

结果：b’Hello tensorfolw’,则安装成功。

2.pytorch安装

退出环境

conda deactivate

查看显卡配置

nvidia-smi

cuda版本记下11.0

激活环境开始安装

conda activate ptts

pytorch官网查下载指令(https://pytorch.org/get-started/locally/)

conda install pytorch torchvision torchaudio
或者
conda install pytorch torchvision cudatoolkit=11【cuda版本】

检验是否成功，成功输出true

import torch
print(torch.cuda.is_available())

查看环境安装包

pip list

3. jupyter notebook使用环境

1. 激活conda环境

 conda activate ptts

2. 安装ipykernel

 conda install ipykernel

3. 将环境写入notebook的kernel中

  python -m ipykernel install --user --name 环境名 --display-name “python 环境名”

  python -m ipykernel install --user --name tensorflow+pytorch --display-name “pyt+tensor”

4. 打开notebook
   在需要的文件夹中 ，打开cmd，输入jupyter notebook
   

成功！