栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

pytorch安装 windows10 64位

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

pytorch安装 windows10 64位

记录一下pytorch的安装过程。

我是参考的Pytorch(一) Pytorch 的安装_开开心心-CSDN博客_pytorch

虽然我的驱动版本很高,我还是选择了像上面链接那样做。

如果你下载cudnn有麻烦,可以试试下一个链接:

Index of /anaconda/pkgs/main/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

注:安装结束后将  ~/nvcc/bin(因为版本的不同可能在不同的地方) 目录添加到环境变量,我是通过搜索nvcc.exe找到的

测试cuddn需要用到的例子在:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1extrasdemo_suite

安装完cuddn之后,想着添加path环境变量:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0libx64(具体路径看你自己的)

安装anaconda(就用不着安装python了,因为anaconda都包含了),可以在下面的链接里去找

Anaconda | The World's Most Popular Data Science PlatformAnaconda is the birthplace of Python data science. We are a movement of data scientists, data-driven enterprises, and open source communities.https://www.anaconda.com/

点击那个get start,我选的最新版本下载。

要想验证成功与否,装完成以后,在最近添加和anaconda包中都能找到anaconda prompt。

如果弹出黑窗口,显示base就证明成功了

安装pytorch,需要在pytorch官网上,做好选项如下:

之后,在上面anaconda命令行中就输入

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

就好了,等待。。。当然,中间需要输入yes or no

下载完成后我们再次输入pip list,查看到已经有torch的存在。

检验安装

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/308466.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号