cuda环境配置

windows cuda10 安装

1. 下载cuda 和cudnn
2. 参照视频
3. 测试: nvcc -V

链接

pip3 install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
# 测试
import tensorflow

tensorflow安装

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.0.0-alpha0

# 测试
import torch
print(torch.__version__)
print(torch.cuda.is_available())