Windows+pycharm安装pytorch

最近要参加一个比赛,需要用到深度学习的框架,老师建议我们用pytorch,但是安装的时候总是发现各种错误。

解决方案是:去pytorch官网生成下载的命令,然后再cmd上执行

Windows+pycharm安装pytorch_第1张图片

这里我们发现有CUDA的选项,这就需要我们安装cuda

怎么安装cuda?

首先我们要知道我们的独立显卡的型号:

Windows+pycharm安装pytorch_第2张图片

然后我们要知道我们的显卡是否支持cuda:点击这里

Windows+pycharm安装pytorch_第3张图片 

 发现是支持的,算力为6.1:

Windows+pycharm安装pytorch_第4张图片

接下来就要下载cuda了,选择10.1(最好)或者10.0都行:点击这里 

Windows+pycharm安装pytorch_第5张图片

 然后下载Windows版本的:

Windows+pycharm安装pytorch_第6张图片

 下载完成后:

 接下来就是安装:默认路径就好

Windows+pycharm安装pytorch_第7张图片

 选择自定义安装:

Windows+pycharm安装pytorch_第8张图片

 按照这个来打勾:

Windows+pycharm安装pytorch_第9张图片

 然后配置环境变量:

Windows+pycharm安装pytorch_第10张图片

然后观察安装是否成功:

Windows+pycharm安装pytorch_第11张图片

CUDNN的配置和下载:点击这里

Windows+pycharm安装pytorch_第12张图片

 然后就需要加入一下,创建个账户,问卷该填的填了:这里我下载的是10.0你们可以根据下载了10.1的cuda就下载10.1的cudnn

Windows+pycharm安装pytorch_第13张图片

 接下来就是下载pytorch了,直接用下面的指令的话速度极慢,还容易断连,所以以下方法抛弃

Windows+pycharm安装pytorch_第14张图片

 

我们去 https://download.pytorch.org/whl/torch_stable.html下载响应的torch文件和torchvision文件,注意版本号,一个是1.4.0,一个是0.5.0

Windows+pycharm安装pytorch_第15张图片

下载好之后:

Windows+pycharm安装pytorch_第16张图片

切到目录下:

Windows+pycharm安装pytorch_第17张图片

 执行相应命令:

Windows+pycharm安装pytorch_第18张图片

 

 最后安装成功后:检验

Windows+pycharm安装pytorch_第19张图片

大功告成!!!!,大家还有什么疑问欢迎交流

 

你可能感兴趣的:(pytorch--深度学习,深度学习,pytorch)