1. 查看cuda版本
nvcc --version
2. 安装 cuda 和 cudatoolkit
conda install cudatoolkit=10.2 cudnn
3. 安装tensorflow (gpu)
pip install tensorflow==2.5.0
4. 安装pytorch (gpu)
[linux] pytorch 安装 torch torchvision gpu 配置_熊熊的小心心的博客-CSDN博客
pip install torch==1.13.1
#conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch