Pytorch Paddle TensorFlow 在一个GPU上运行

借助Anaconda，可以设置不同的环境 cuda-toolkit cudnn匹配网址 https://developer.nvidia.com/rdp/cudnn-archive

1. 安装Anaconda

https://www.anaconda.com/products/distribution

安装完成后，在windows下可以打开 Anaconda Prompt （可以用管理员运行，可能有部分库需要管理员权限）

2.1 安装Pythorch

参考 https://pytorch.org/

conda crea

Read more

1. 流程

1.1 打开torch官网

https://pytorch.org/get-started/locally/

1.2 上图会告诉下载的Cuda tookit 版本，以及如何下载torch，torchvision

1.3 安装cuda tookit

https://developer.nvidia.com/cuda-11.3.0-download-archive

1.4 安装torch，torchvision

pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f

Read more