首页 > 其他分享 >PyTorch 安装

PyTorch 安装

时间:2022-09-06 10:24:38浏览次数:85  
标签:torch 环境 PyTorch conda Anaconda 安装

  • 本机环境

打开设备管理器,找到相应的显卡,查看本机是否支持GPU,如下图所示,这可以确定需要安装PyTorch 是GPU 还是CPU 版本

  • Anaconda 安装

Anaconda官网地址:https://www.anaconda.com/distribution/,选择和机器环境对应的版本并下载安装

 

  • PyTorch 安装

PyTorch 主页:https://pytorch.org/

PyTorch 主要使用Pip 或 Conda安装,这里使用Anaconda作为安装工具,选择CUDA版本

 

打开Anaconda,创建一个新的环境,指定Python的版本

 

 

 

 

打开Anaconda Prompt,可以查看当前的环境或者切换环境:

 

 

 

 

列出环境:conda info --env

切换环境:conda activate My_PyTorch

查看安装的包:conda list

 

使用Pip方式安装,命令:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113,执行过程如下

 

  • 验证PyTorch是否安装成功

1:打开python:python

2:测试pytorch:import torch

如下图所示,没有遇到问题,则表示成功了

 

 

  • 安装Jupyter

安装Anaconda的同时,会在base环境安装jupyter notebook,因此并不能在此jupyter notebook运行torch包,在前面新建的环境中运行命令

conda config --set remote_read_timeout_secs 1000.0

conda install nb_conda

 

 

完成后,执行jupyter notebook,如下图,选择之前创建的conda 环境

 

输入import torch,点击运行
再输入torch.cuda.is_available()
输出 True,表示我们在自己创建的pytorch环境中安装好了jupyter包

 

标签:torch,环境,PyTorch,conda,Anaconda,安装
From: https://www.cnblogs.com/benfly/p/16379433.html

相关文章

  • Linux 安装telnet
    一、安装telnet1、首先我们检测telnet-server的rpm包是否安装[root@localhost~]#rpm-qatelnet-server若无输入内容,则表示没有安装。linux的telnet-server.rpm默认......
  • ubuntu 22.04 安装gcc-7、g++-7
    ubuntu22.04安装gcc-7、g++-71、安装gcc-71.1添加镜像(base)root@FIB:~#vim/etc/apt/sources.list(base)root@FIB:~#tail-1/etc/apt/sources.listdeb[arch......
  • swoole 安装和简单实用
    Swoole是:PHP语言的高性能网络通信框架,提供了PHP语言的异步多线程服务器,处理异步任务,异步TCP/UDP网络客户端,异步MySQL,数据库连接池,AsyncTask,消息队列,毫秒定时器,异步文件读写......
  • 二进制部署k8s集群v1.23.9版本-17-安装minio
    17.1、准备minio镜像192.168.1.200操作dockerpullminio/minio:latestdockerimages|grepminiodockertage31e0721a96bharbor.qgutech.com/middleware/minio:lates......
  • python3下载及安装教程(Windows)
    Python目前已支持所有主流操作系统,在Linux,Unix,Mac系统上自带Python环境,一般默认装的是Python2版本,Windows系统上没有Pyhton环境,需要我们手动安装一下,现在一般都是python3......
  • pytorch-is_leaf-叶子张量
    1概念介绍        在pytorch的tensor类中,有个is_leaf的属性,表示这个tensor是否是叶子节点:is_leaf 为False的时候,则不是叶子节点, is_leaf为True的时候为叶......
  • Linux MiniMal版本常规所需环境安装
    环境说明:开始之前建议先拍摄快照、必免不必要的丢失VMware®Workstation16ProCentos7.9minimal版本(安装完成并未做任何操作)......
  • Kafka Docker安装
    安装Zookeeperdockerrun--namezookeeper-d-p2181:2181-v/etc/localtime:/etc/localtimelibrary/zookeeper安装Kafkadockerrun-d--namekafka-p9092:9092......
  • 安装程序无法自动安装 Virtual Machine Communication Interface Sockets (VSock) 驱
    安装程序无法自动安装VirtualMachineCommunicationInterfaceSockets(VSock)驱动程序,必须手动安装此驱动程序遇到的问题:虚拟机中安装的win7系统,在安装VMtools的时......
  • 性能测试工具JMeter(一)---安装与运行
    JMeter简介1、Apache组织开发的开源免费压测工具2、纯java程序,跨平台性强3、源码可以从网上下载4、高可扩展性5、可对服务器、网络或对象模拟巨大的负载,进行压力测......