1.安装tensorflow
pip install tensorflow==2.0.0-alpha0
2.安装pytorch
2.1先配置清华源(可能一些网址已经有)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
https://blog.csdn.net/weixin_35057064/article/details/105562493 中科大源
2.2再安装pytorch
conda install pytorch torchvision cudatoolkit=10.1 pytorch
2.3测试(import 没问题就说明成功了)
import torch
2.4使用自己环境下的pytorch
import sys
sys.path
#会看到自己环境下的pytorch在哪
sys.path[0] = '/home/ubt/anaconda3/envs/zdy/lib/python3.7/site-packages'
执行上面的代码之后就可以了
import torch
torch.__file__