创建新的环境 conda create -n pythorch_tensorflow_paddle python=3.8
conda install paddlepaddle-gpu==2.5.1 cudatoolkit=11.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/Paddle/ -c conda-forge
验证:
import paddle
paddle.utils.run_check()
如果出现PaddlePaddle is installed successfully!,说明您已成功安装。
pip install tensorflow-gpu==2.11.0
验证:
import tensorflow as tf
# print(tf.test.is_gpu_available()) # ---- tf 1.x 版本测试
print(tf.config.list_physical_devices("GPU")) # ---- tf 2.x 版本测试
tf.test.is_gpu_available()
pip install torch
验证:
import torch
print(torch.cuda.is_available())