Windows
查看 CUDA 版本
在命令行中执行:nvcc --version
查看 CUDNN 版本
进入 CUDA 的安装目录
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include
用记事本方式查看文件 cudnn.h,如下所示,CUDNN 版本为 7.2.1
pytorch下载
地址:
https://download.pytorch.org/whl/cpu/torch_stable.html # CPU-only build
https://download.pytorch.org/whl/cu80/torch_stable.html # CUDA 8.0 build
https://download.pytorch.org/whl/cu90/torch_stable.html # CUDA 9.0 build
https://download.pytorch.org/whl/cu92/torch_stable.html # CUDA 9.2 build
https://download.pytorch.org/whl/cu100/torch_stable.html # CUDA 10.0 build
https://download.pytorch.org/whl/torch_stable.html
https://www.lfd.uci.edu/~gohlke/pythonlibs/
镜像:
https://pypi.doubanio.com/simple/
https://pypi.tuna.tsinghua.edu.cn/simple/
pytorch对应的torchversion
08-10
288
08-13
2015