win10 + cuda9.0+pytorch环境配置

  • 默认nvidia显卡驱动安装成功
  • CUDA:
    • 查看版本号:https://www.jianshu.com/p/d3b9419a0f89
    • CUDA Toolkit Archive : https://developer.nvidia.com/cuda-toolkit-archive
    • nvcc -V
  • cudnn:
    • https://developer.nvidia.com/rdp/cudnn-archive
    • 复制bin、include、lib这三个文件夹,粘贴到cuda的路径“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1”下即可。
    • (3)将bin所在的目录添加到环境变量 PATH 中,“此电脑”→“高级系统设置”→“环境变量”→“系统变量”→“path”→“编辑”→“新建”加入该路径即可。
  • pytorch
    • 创建虚拟环境 conda create -n … python=3.6 pip ipython
    • https://discuss.pytorch.org/t/install-pytorch-in-windows10/17056/4
      pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
      pip install torchvision

你可能感兴趣的:(环境安装)