准备工作
cuda11.0+cudann8.0.4+anaconda
1. 创建conda环境
conda create --name NewPytorch python=3.7
进入
activate NewPytorch
2.安装Pytorch
pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
下载速度很理想,五分钟即可完成
安装成功
3.安装Tensorflow
安装cudatoolkit11
conda install cudatoolkit=11
安装Tensorflow2.4
直接下载好本地安装