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())