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()
©️2020 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值