Windows下用Anaconda安装Pytorch/torchvision

1. 安装Anaconda最新版

2. 依赖的环境

    Anaconda3 x64 (with Python 3.5/3.6)
    Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
    CUDA 8 / CUDA 9
    cuDNN v5以上
    如果安装了CUDA编译的包,请确保你的电脑有Nvidia的显卡。
    注:这里没有介绍GPU版本的安装方法,如需要的请搜索其他博文。

安装CUDA 8

(1)首先在官网上(https://developer.nvidia.com/cuda-downloads)下载CUDA,选择自己合适的版本。
该链接界面只显示最新版本。若需要下载以前的版本,可在页面下方点击Legacy Releases,选择自己需要的其他版本,这里安装的是cuda8.0。
这里写图片描述

在Windows中使用迅雷下载,这样比较快。

(2)根据相应的系统选项,我选择的是cuda_8.0.61_win10.exe(大小为1.3G),安装的时候建议选择 自定义 而不是“精简”(从下面的英文解释可以看出,其实这里的精简写成完整应该更贴切,他会安装所有组件并覆盖现有驱动,然而我并不想安装全家桶,何况我的官方显卡驱动比他的新)。

双击安装下载的.exe文件,然后选择解压路径,如下图,解压到哪里无所谓,安装成功会自动删除;

解压完成后,得到如下图:

  • 精简:安装所有CUDA模块,并覆盖掉当前的NVIDIA驱动程序;(说实话,容易出问题)
  • 自定义:选择自己想要安装的模块,此处选择这里;

选择自定义后,出现下图所示:

下面几个模块准确具体有什么用,不能100%确定,但能大概才出来:

  • CUDA:这个是必须的,下面有CUDA Runntime、Samples一些东西;
  • NVIDIA GeForce Experience:这个好像是为了更好的游戏体验,之前安装显卡驱动程序时也提示是否安装,果断拒绝了;
  • Other components:这里的PhysX好像也是为了游戏体验来的;
  • Driver components:这个就要慎重了,意思就是重新安装显卡驱动程序;如果之前已经成功安装驱动程序,这里就不用选了;如果之前没安装驱动程序,建议还是去官网上单独下载驱动程序进行安装吧;

选择好需要安装的模块,这里我都选了,但是第一项cuda中的 Visual Studio 的选项,不用的话,好像就不用安装。

选择安装路径了,我的选择如下图:

  • 在如图所示位置建立相应文件夹,然后再指定安装路径;

安装成功后;Ctrl+R,打开cmd:

nvcc -V

输出版本信息,则表明安装成功。

配置环境变量

F:\CUDA\NVIDIA GPU Computing Tookit\CUDA\v8.0\lib\x64添加的系统变量的path中;

2. 安装CuDNN

下载对应CUDA 8.0版本的CuDNN:下载链接

(如果安装的是其他版本的CUDA,注意CuDNN的版本)

下载完成后,解压得到一个名为cuda的文件夹;将该文件夹下的文件复制到上一步安装的CUDA中;注意对应的文件夹;

./cuda/bin/**.dll 复制到  ./NVIDIA GPU Computing Tookit/CUDA/v8.0/bin/
./cuda/include/**.dll 复制到  ./NVIDIA GPU Computing Tookit/CUDA/v8.0/include/
./cuda/lib/x64/**.dll 复制到  ./NVIDIA GPU Computing Tookit/CUDA/v8.0/lib/x64/

安装完成;

3. 测试1

使用VS2013打开./cuda/v8.0/Samples_vs2013.sln;

加载完成后,执行本地Windows编译,最后输出成功,见下图;

 

3. 开始菜单打开Anaconda Prompt,在里面输入conda create -n pytorch python=3.5,为pytorch创建一个虚拟环境

 

 


安装cuda参考:

           https://blog.csdn.net/weixin_34409703/article/details/93226830

           https://blog.csdn.net/yimingsilence/article/details/79126914

           https://blog.csdn.net/zhuiqiuzhuoyue583/article/details/88756053

安装Pytorch/torchvision参考:

          https://blog.csdn.net/yimingsilence/article/details/79126914

          https://blog.csdn.net/Tan915730/article/details/78954482

          https://blog.csdn.net/zhuiqiuzhuoyue583/article/details/85327559

          https://blog.csdn.net/hpuspring/article/details/79756211

          https://zhuanlan.zhihu.com/p/37748231

          

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章