由于最近在搞一些强化学习的项目,编程中又需要用到Pytorch,而且我自己比较喜欢使用Visual Studio进行编程,所以钻研了一下怎么在Visual Studio上把Pytorch跑起来,说实话这个过程还挺复杂的,搞了半天才搞定,就想把操作的过程写出来,供大家参考。
Visual Studio: 我使用的是Visual Studio 2017,不过应该别的版本都没所谓,由于是用Python进行开发,所以在Visual Studio的开发工具处要把Python开发勾选上,并勾选Anaconda3


步骤一:首先去到NAVIDA下载CUDA的官方网站(https://developer.nvidia.com/cuda-downloads),选择适合的版本进行下载,笔者使用的是Windows10操作系统,所以如图示勾选

步骤一:在NAVIDA的官方网站(https://developer.nvidia.com/rdp/cudnn-archive),选取与你CUDA版本对应的CUDNN下载即可,不过下载CUDNN需要注册,会需要填写一些信息,这个按照流程走就好






这个地方坑有点多,待笔者细细道来





接下来,我去请教了一下其他同学,并且也查了一些资料,发现可以在清华镜像网站上直接安装,速度提升了很多,附上网址:https://mirror.tuna.tsinghua.edu.cn/help/anaconda/ 点进去下拉,在页面底端找到Pytorch的指令

步骤二:在PowerShell中输入正确的指令进行安装
首先将上面那段换源指令输入PowerShell
再输入Pytorch官网上对应版本的安装指令:
PS:注意,这里要把后面的-c Python去除,不然仍然会用官方仓库进行安装,速度仍然上不去



如果显示结果为“True”,则说明安装成功
花了一番功夫,总算是把Pytorch安装成功了,其实计算机的很多技术知识都是这样,说它简单也不简单,说它难也不难,但只要你有恒心,肯钻研,多查资料,吸取他人的经验,总能把问题钻懂

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容,请联系我们,一经查实,本站将立刻删除。
如需转载请保留出处:https://51itzy.com/kjqy/199747.html