cuda pytorch 环境变量_win10+CUDA+pytorch机器学习的第一步安装配置环境!

pytorch和tensorflow用很多人要问,如果你是工程师更推荐使用TensorFlow2.0 如果你想从事科研学术或者学生可以选择pytorch可以满足快速迭代和发表文章的需求。

安装配置环境是劝退的第一步,按着下面步骤一定可以!

现在来说在win10系统下 pytorch的下载安装:

1. 安装anaconda3(很多人推荐下载安装但是我在anaconda中下载pytorch失败了,下面步骤成功)

2. 安装CUDA:

需要注册,下载时间长

2.2环境变量配置

系统属性环境变量里新增5个变量

CUDA_BIN_PATH %CUDA_PATH%\bin

CUDA_LIB_PATH %CUDA_PATH%\lib\Win64

CUDA_SDK_BIN %CUDA_SDK_PATH%\bin\Win64

CUDA_SDK_LIB %CUDA_SDK_PATH%\common\lib\Win64

CUDA_SDK_PATH C:\cuda\cudasdk\common

2.3测试是否配置成功

Cmd中输入:nvcc -V 出如下图配置成功

3、安装pytorch

3.1安装

复制run this command中的内容在cmd中下载

安装成功CUDA之后再安装pytorch非常快

3.2最后检测下是不是安装成功

在python环境中 cmd输入python

之后输入import torch没有报错就没有什么问题

接下来,输入 torch.cuda.is_available,如果没有问题,意味可以使用 GPU

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值