【安装库】安装虚拟环境pytorch or tensorflow..

conda env list

2.创建虚拟环境

conda create -n tf_gpu python=3.6.5

3.激活虚拟环境

conda activate tf_gpu

4.安装pytorch

pip install torch==1.10 -i https://pypi.tuna.tsinghua.edu.cn/simple --target=E:\software\Anaconda\run_file\envs\pytorch_1.10\Lib\site-packages

或者

pip install torch===1.3.0 torchvision===0.4.1 -f https://download.pytorch.org/whl/torch_stable.html

5.cuda安装
选择对应版本