创建环境:
conda create -n torch python=3.7
需要先安装cuda(只需要安装cudatoolkit)、cudnn
需要指定版本,否则会因为版本不对应而不能使用
conda install cudatoolkit=10.0.130
conda install cudnn=7.6.0
(安装ten-gpu 需要制定镜像,速度飞起)
pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
版本对应(keras自动下载):
cudatoolkit 11.3.1 h59b6b97_2 https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main
cudnn 8.2.1 cuda11.3_0 https://mirrors.sjtug.sjtu.edu.cn/anaconda/pkgs/main
keras 2.9.0 pypi_0 pypi
keras-preprocessing 1.1.2 pypi_0 pypi
pytorch 1.11.0
tensorflow-gpu 2.6.0 pypi_0 pypi
ten1版本:
cudatoolkit 10.0.130 0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
cudnn 7.6.0 cuda10.0_0 https://mirrors.ustc.edu.cn/anaconda/pkgs/main
tensorflow-gpu 1.14.0 pypi_0 pypi