anaconda同一个虚拟环境安装pytorch和tensorflow

anaconda下同一个虚拟环境安装pytorch和tensorflow

最近在重装一个硬盘，为了方便，希望能在一个conda环境下使用pytorch和tensorflow-gpu，由于pytorch和tensorflow-gpu都对cuda的版本有不同的对应关系，需要寻找相同cuda版本下的pytorch和tensorflow-gpu。例如：

conda install tensorflow-gpu=2.4.1

conda命令能装的目前貌试最高到2.4.1，对应的cuda为10.1.安装完成后进入环境,检查是否可用。

import tensorflow as tf
tf.test.is_gpu_available()

然后，根据其cuda版本寻找对应的pytorch

如图，选择第二条的版本
# conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
输入对应指令完成安装。检测pytorch是否安装成功