栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Pytorch环境配置

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Pytorch环境配置

  1. 使用conda创建pytorch环境
conda create -n pytorch1.2.0 python=3.6

其中pytorch1.2.0是要创建的环境名称,可随意设置,python=3.6代表要使用的python版本。示意如下,输入y进行创建。
创建完成。

  1. 激活pytorch环境,进行环境配置。
conda activate pytorch1.2.0

括号内环境变为pytorch代表激活成功。

  1. 安装pytorch。
    进入pytorch官网https://pytorch.org,找到对应版本的pytorch和gpu进行安装。这里选择安装cuda9.2版本。
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

如嫌安装速度太慢可用清华镜像源进行安装。
添加清华镜像源。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

查看添加的镜像源。

conda config --set show_channel_urls yes | cat ~/.condarc

然后再进入pytorch官网找到对应版本进行下载,需去掉指令中的-c pytorch,如下。

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2
  1. 安装完成检查是否安装成功。

先在环境中输入python,然后输入如下代码。

import torch

显示如下图说明安装成功。

  1. 检测是否可在GPU进行使用。
torch.cuda.is_available()

出现True说明可以在GPU上使用。

安装完成!
可用pip list指令检查安装的包。

如下载需要的包可用清华源进行加速下载。

pip install XXX -i https://pypi.tuna.tsinghua.edu.cn/simple
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/864426.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号