cuda环境配置

windows cuda10 安装

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

pytroch 安装链接

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())
posted @ 2019-08-12 13:35  FromZeroToOne  阅读(1020)  评论(0)  编辑  收藏  举报