Cuda

GUP加速器的驱动

到 Official website 选择合适版本驱动下载并安装

针对神经网络的加速库

到 cudnn 选择合适版本下载，并按 官方指示

Pytorch

Official website

installing

1、Try

pip install torch torchvision

2、If failed, do

文件下载地址：https://download.pytorch.org/whl/torch_stable.html

下载适合你的 torch 和 torchvision 版本 ，然后离线安装

pip install xx.whl

Verifying

import torch
x = torch.rand(5, 3)
print(x)

有输出就安装成功了

import torch
print(torch.cuda.is_available())

True代表 GPU可以用了

Tensorflow

installing

pip install tensorflow-gpu==2.0.0-rc0

Verifying

import tensorflow as tf
tf.enable_eager_execution()
print(tf.add(1, 2).numpy())
hello = tf.constant('Hello, TensorFlow!')
print(hello.numpy())

keras

pip install keras