Windows系统下,深度学习CUDA和pytorch的安装

 近期有需要用到pytorch的环境,所以就去安装了相应的CUAD和pytorch,现在做一个简单的流程介绍

1.CUDA的安装

2.安装pytorch


1.CUDA的安装

对于这个的安装,我主要是借鉴https://zhouchen.blog.csdn.net/article/details/107716721?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-3.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-3.control这篇博客,可以根据上面的步骤一步一步来,本人安装的是CUDA10.2版本,但我到最后一步

并没有出现如上图所示的画面,而是一闪而过,所以我就去 cmd中输入了

nvcc -V

出现了CUDA相应的版本号,应该也算是安装成功了。

 2.安装pytorch

先去anaconda3中创建一个pytorch的虚拟环境(anaconda3的安装步骤这里不多赘述),有两种方法可以进行虚拟环境的创建,

1.直接进入anaconda3中,点击Environments,可以看见你现有的虚拟环境,然后点击左下角的create按钮,就可以进行虚拟环境的创建,如图所示,然后填入环境name进行创建。

 2.也可以通过anaconda prompt中输入“conda create -n pytorch36 python=3.6”进行创建,该命令中pytorch36就是虚拟环境的name,python的版本是3.6.

虚拟环境创建完毕,然后进入pytorch官网,https://pytorch.org.找到自己所需的pytorch版本。

tips:需要找到自己所安装的CUDA相应的版本的pytorch。

 

点击Get Started,可以看见最新的版本号,然后点击Previous PyTorch Versions可以查看历史版本。

 复制所选中的那句话,然后进入anaconda prompt中,首先激活你之前所创建的虚拟环境,

conda activate xxxxxx

xxxxx对应的是你虚拟环境的name,成功的话是会将前面括号里面的(base)改成(环境name)

然后将复制的语句粘贴进去,正常情况下会进行安装pytorch,但是我自己安装的时候出现了问题,PackagesNotFoundError: The following packages are not available from current channels 

然后发现在该情况下输入

conda config --add channels conda-forge

该指令,然后粘贴之前的pytorch安装命令就可以正常安装pytorch

Proceed 选择“y”,然后就进行下载以及安装,上图是在下载相应的包。安装完毕后,在后面输入python

 出现版本号,然后分别输入

import torch


torch.cuda.is_available()

出现true就说明 机器的GPU就可以被Pytorch使用。

也就可以使用该pytorch了!!!

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值