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是否安装成功