Pytorch 学习笔记(1)-安装

我的github地址:https://github.com/poptree

1. 安装环境

推荐使用anaconda对环境进行管理,这里以anaconda3.5.1,ubuntu为例,安装步骤如下:
1. 下载anaconda

wget https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh
  1. 安装anaconda
bash Anaconda3-5.1.0-Linux-x86_64.sh

接下来根据提示安装即可。
anaconda在安装完成后需要重启终端。
其他anaconda使用方法在这里不在赘述。

2. 安装Pytorch

推荐从Pytorch官网下载Pytorch。同时官网也提供了使用conda命令安装Pytorch的命令。
这里以conda,python3.6,cuda8为例子。

conda install pytorch torchvision -c pytorch

等待安装完成后,激活conda环境即可使用Pytorch。
运行以下代码即可查看是否安装成功。

import torch
torch.cuda.is_available()
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值