pytorch安装详细步骤

人工智能27

文章目录

(一)win—配置tensorflow-GPU

直接查看这条链接即可:win-配置tf-GPU
本人用的conda和tensorflow-GPU版本下载:提取码:98ot
环境:win10+anaconda
注:anaconda安装步骤略,以下步骤默认anaconda已安装。

(二)安装 pytorch

2.1 创建虚拟环境


pytorch安装详细步骤


conda create --name pytorch python=3.8.1

注意,这里的 pytorch 是虚拟环境的名称,可随意取。3.8.1是我机器上的python版本,可结合自己安装的python版本灵活变换。


activate pytorch

pytorch安装详细步骤

2.2正式安装pytorch

****:pytorch 官网链接
pytorch安装详细步骤
注意:如果笔记本有独立显卡(NVIDIA)的话,可以如上选择对应的CUDA版本,否则选择CPU。

  • 本人采用的方法
    在pytorch虚拟环境下,创建.condarc文件
    pytorch安装详细步骤
    在虚拟环境中输入如下命令:
conda config --set show_channel_urls yes
  • 之后在自己电脑的用户文件下找到一个.condarc文件
    pytorch安装详细步骤
    用记事本打开这个文件,然后用如下代码代替其中的内容:
channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

记住.condarc配置好后一定要保存。

  • 或者使用清华镜像源
    在浏览pytorch安装帮助的相关帖子时有人说清华源停止镜像了,但是现在清华源已经恢复提供镜像了,所以还是可以用的。附上清华源Anaconda镜像使用帮助,大家可以读一下这段帮助(不读也没关系下面给出详细步骤)。
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

复制这条命令在 Anaconda Prompt 里输入。

2.3 验证是否安装成功

是否安装成功分两个方面。

  • 一个是在prompt里面

(1)在命令行左边为 pytorch 环境中,输入 python

(2)之后,输入 import torch,如果没有报错,意味着 PyTorch 已经顺利安装了。

  • 一个是在jupyter notebook里面调用
    首先,在菜单中打开Anaconda Prompt,然后安装插件。
    命令行输入:
conda install nb_conda

然后进入创建的pytorch环境,命令行输入:

conda install ipykernel

安装成功的样子如下:
pytorch安装详细步骤
进入Anaconda中可查看,多了pytorch框架:
pytorch安装详细步骤

(三)本文参考链接如下:(感谢各位大佬)

WIN10下pytorch环境配置(安装了半天的血泪史)
WIn10+Anaconda环境下安装PyTorch(避坑指南)
win10下使用anaconda安装pytorch(清华镜像)
如何让Jupyter Notebook支持pytorch

Original: https://blog.csdn.net/weixin_54546190/article/details/120754242
Author: ☞源仔
Title: pytorch安装详细步骤

相关文章
计算机视觉OpenCV-图像直方图 人工智能

计算机视觉OpenCV-图像直方图

💥💥💥 欢迎来到本博客💥💥💥 🎉 作者简介:⭐️⭐️⭐️ 目前计算机研究生在读。主要研究方向是人工智能和群智能算法方向。目前熟悉python网页爬虫、机器学习、计算机视觉(OpenCV)、群智能算法。...
librosa能量_librosa语音信号处理 人工智能

librosa能量_librosa语音信号处理

librosa是一个非常强大的python语音信号处理的第三方库,本文参考的是librosa的官方文档,本文主要总结了一些重要且常用的功能。 先总结一下本文中常用的专业名词:sr:采样率、hop_le...
阿里云医疗实体关系抽取大赛 人工智能

阿里云医疗实体关系抽取大赛

1、本项目是基于阿里云比赛开放的医疗数据集去做的实体关系抽取。下面会从数据的详情,模型的选取,模型的训练,模型的验证和模型的预测去讲述。 2、数据准备阶段 1、数据来源是阿里云医疗大赛,选取的是其中一...