人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

这篇具有很好参考价值的文章主要介绍了人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

作者介绍

孟莉苹,女,西安工程大学电子信息学院,2021级硕士研究生,张宏伟人工智能课题组。
研究方向:机器视觉与人工智能。
电子邮件:2425613875@qq.com

本教程提供需要安装的CUDA11.3、Pytorch1.10.0、torchvision0.11.0的安装包,在下述百度网盘链接中自取!

链接:https://pan.baidu.com/s/18mgO8GtW1UnG6uijnnQvIQ?pwd=0843 提取码:0843
–来自百度网盘超级会员V5的分享

一. 查看自己电脑的CUDA版本

根据以下两种方式,查看自己的电脑是否支持CUDA,如果不支持,那么直接跳到[ 七 ]!

1.1 方式一

1.1 .1 使用快捷键‘win + R’- - -> 输入 cmd - - -> 点击确定
人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

1.1.2 在终端输入指令‘nvidia-smi’ - - -> 查看自己电脑支持的CUDA版本

当前CUDA版本为11.6,表示支持最高版本的CUDA是11.6,向下兼容

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

1.2 方式二

1.2.1 在桌面右击 - - -> 打开NVIDIA 控制面板

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

1.2.2 在帮助下 - - -> 选择系统信息

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

1.2.3 在组件 - - -> 查看CUAN版本

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

二. 下载安装CUDA

NVIDIA官网: https://developer.nvidia.com/cuda-toolkit-archive

  • 在NVIDIA官网找到适合自己电脑的CUDA版本,这里建议在满足要求的情况下安装CUDA11.3;本教程提供CUDA11.3版本的安装包
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
  • 选择对应版本的CUDA后进行下载安装;
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
  • 下载好之后安装在默认路径下即可;

CUDA安装参考帖子:https://blog.csdn.net/m0_45447650/article/details/123704930/.

三. 查看环境变量

1. 在设置中–>搜索高级系统设置–>查看环境变量

  • 在环境系统中应该出现以下四个环境变量:
    (该环境变量是CUDA11.3版本的,其他版本的需要和自己下载的CUDA版本相对应)

CUDA_PATH
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3

CUDA_PATH_V11_3
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3

NVCUDASAMPLES_ROOT
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.3

NVCUDASAMPLES11_3_RO…
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.3

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

  • 若没有相应的环境变量需要手动添加

验证CUDA是否安装成功

  • 运行cmd,输入 ‘nvcc -v’ 即可查看版本号;

  • 输入 ‘set cuda’ ,可查看 CUDA 设置的环境变量
    出现下图所示则表明成功安装CUDA
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
    CUDA安装参考帖子:https://blog.csdn.net/m0_45447650/article/details/123704930/.

四. 创建虚拟环境

4.1 使用指令创建虚拟环境

  • 添加镜像源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda config --set show_channel_urls yes
conda config --set ssl_verify false
  • 创建虚拟环境,本次教程安装Python3.6版本;本地cmd终端运行如下指令:
conda create -n py36 python=3.6

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

4.2 查看Anconda中是否出现创建好的虚拟环境

打开Anconda,点击Environments查看是否有名为py36的虚拟环境

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
: 接下来在名为py36的虚拟环境中安装各种科学计算包和框架(Pytorch),若有多个环境,那么例如需要python3.7,就创建python3.7的虚拟环境,有conda管理不会出现版本冲突。

4.3 虚拟环境相关指令

a : 终端运行如下指令:查看本地环境

conda env list

b: 终端运行如下指令:进入虚拟环境

conda activate py36

c: 终端运行如下指令:退出虚拟环境

conda deactivate

注意事项:进入虚拟环境如图:会有小括号(虚拟环境名:py36),进入虚拟环境后就可以在Python为3.6的环境下配置 Pytorch了,尽情安装吧!

五. 安装GPU版本的Pytorch和torchvision

本教程安装CUDA11.3、Pytorch1.10.0、torchvision0.11.0

安装 Pytorch 和 torchvision 分为两种方式,在线方式和离线方式;若在线方式在安装过程中出现安装不上的问题,则可选择离线安装,会提供相应的安装包。

5.1 在线安装

  • 安装镜像源:依次执行下面五条指令;
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
conda config --set show_channel_urls yes
conda config --set ssl_verify false
  • 在本地终端进入创建好的虚拟环境(py36),当最前面出现(py36),则表明已进入py36的虚拟环境;人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
    - 进入Pytorch官网:https://pytorch.org/

  • 点击以往Pytorch版本
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

  • 找到 Windows 系统下 CUDA11.3 下的 Pytorch 安装指令,并复制;
    若安装的CUDA版本不是11.3,则需要选择对应版本的 Pytorch 和 torchvision
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

  • 粘贴在cmd本地终端的py36虚拟环境中运行安装:

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

5.2 离线安装

5.2.1 下载安装torch

Pytorch下载链接:https://download.pytorch.org/whl/torch_stable.html.

  • 选择对应版本的torch并下载
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

  • 在本地终端输入以下指令进行离线安装,当出现 Successfully installed… 则表明安装成功;使用下面三个中的任意一个指令即可;

pip install F:\Demo\torch\torch-1.10.0+cu113-cp36-cp36m-win_amd64.whl
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple F:\Demo\torch\torch-1.10.0+cu113-cp36-cp36m-win_amd64.whl
pip install F:\Demo\torch\torch-1.10.0+cu113-cp36-cp36m-win_amd64.whl http://pypi.douban.com/simple/trusted host pypi.douban.com

备注:上述指令中的 “F:\Demo\torch\torch-1.10.0+cu113-cp36-cp36m-win_amd64.whl” 表示下载torch安装包并保存到自己电脑上的路径 + 文件名;根据自己下载保存的情况进行更改;

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

5.2.2 下载安装torchvision

Pytorch、torchvision和python具有版本对应关系

  • Pytoch 和 torchvision 的版本对应关系可在Pytorch官网查询,如下:
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

本教程查到 Pytoch1.10.0 对应的 torchvision 版本为 0.11.0

  • 离线下载torchvision 0.11.0
    torchvision下载链接:https://download.pytorch.org/whl/torch_stable.html.

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

  • 使用 pip 指令安装torchvision0.11.0;使用下面三个中的任意一个指令即可;
pip install F:\Demo\torchvision\torchvision-0.11.0+cu113-cp36-cp36m-win_amd64.whl
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple F:\Demo\torchvision\torchvision-0.11.0+cu113-cp36-cp36m-win_amd64.whl
pip install F:\Demo\torchvision\torchvision-0.11.0+cu113-cp36-cp36m-win_amd64.whl http://pypi.douban.com/simple/trusted host pypi.douban.com

备注:上述指令中的 “F:\Demo\torchvision\torchvision-0.11.0+cu113-cp36-cp36m-win_amd64.whl” 表示 torchvision 安装包的路径+文件名;根据自己的情况进行更改;

- 若在安装torchvision的过程中出现自动更新torch的情况,则使用下述指令安装torchvision;

pip install --no-deps F:\Demo\torchvision\torchvision-0.11.0+cu113-cp36-cp36m-win_amd64.whl

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

5.3 验证是否成功安装

  • 安装完成后,在虚拟环境中依次运行如下指令进行验证Pytoch是否安装成功以及GPU是否可用;
python
import torch 
torch.cuda.is_available()

人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

若 import torch 不报错,说明 torch 安装成功

若 torch.cuda.is_available() 结果为True,则表示GPU版本安装完成

六. 在VsCode 中使用虚拟环境

6.1 安装Python插件

  • 进入VsCode软件,点击“扩展”,并在搜索框中输入Python,然后选择Python插件右下角的Install;
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

6.2 添加虚拟环境

  • 按快捷键 “Ctrl+Shift+P” ,调出全局设置搜索窗口,然后输入 “Python:Select Interpreter” 后会出现 “Python:Select Interpreter” 选项,点击该选项;
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

  • 点击该选项会后跳转到 Python 解释器配置窗口,这里显示的是已经添加好的Python解释器,选择py36,即可在VsCode中使用Python了;
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

  • 注意事项:若VsCode进入终端后是PS环境;则先在终端中输出cmd,退出PS环境,再手动输入 “conda activate py36” 进入配置好的py36环境中; 通过输入“pip list” 即可查看安装好的 torch 和 torchvision,确认是自己下载安装的版本即可;
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境

以上就是关于在 Anaconda 中创建虚拟环境、安装GPU版本的Pytorch及torchvision、以及在VsCode中使用虚拟环境的全部内容!!!

接下来是CPU版本的Pytorch的安装教程!!!

前提:先完成步骤四.创建虚拟环境

七. 安装CPU版本的Pytorch和torchvision

  • 进入虚拟环境,使用Pytoch官网根据指令安装运行即可;
    人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cpuonly -c pytorch
  • 验证是否安装成功,import torch不报错就成功了:
python 
import torch 

以上就是关于在 Anaconda 中创建虚拟环境、安装CPU版本的Pytorch及torchvision的全部内容!!!文章来源地址https://www.toymoban.com/news/detail-431548.html

到了这里,关于人工智能之配置环境教程二:在Anaconda中创建虚拟环境安装GPU版本的Pytorch及torchvision并在VsCode中使用虚拟环境的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 🔥🔥探索人工智能的世界:构建智能问答系统之环境篇

    在之前的一篇文章中,我们已经对项目所需的知识点进行了简单的了解。今天,我们将开始搭建整个项目的环境。在接下来的文章中,我们将详细介绍如何配置项目所需的各种工具和环境,以确保项目的顺利进行。 接下来,你可能需要多次重启电脑来确保环境变量的生效。这

    2024年02月05日
    浏览(23)
  • 人工智能教程(二):人工智能的历史以及再探矩阵

    目录 前言 更多矩阵的知识 Pandas 矩阵的秩 在上一章中,我们讨论了人工智能、机器学习、深度学习、数据科学等领域的关联和区别。我们还就整个系列将使用的编程语言、工具等做出了一些艰难的选择。最后,我们还介绍了一点矩阵的知识。在本文中,我们将深入地讨论人

    2024年02月05日
    浏览(14)
  • 【AI】人工智能本地环境集成安装

    目录 1、基础安装 1.1 GPU安装 1.1.1 GPU版本支持 1.1.2 下载CUDA 1.1.3安装CUDA 1.1.4配置环境变量 1.1.5检测CUDA是否安装成功  1.2 CUDNN安装 1.2.1 下载CUDNN 1.2.2 添加配置 1.2.3验证结果 2、pytorch安装

    2024年02月03日
    浏览(59)
  • 人工智能与智能家居:创造更舒适的生活环境

    随着科技的发展,人工智能(AI)已经成为了我们生活中不可或缺的一部分。智能家居是人工智能在家庭生活环境中的一个重要应用领域。智能家居可以通过集成多种传感器、摄像头、声音识别等设备,实现智能化的控制和管理,从而提高生活质量,提高生产效率,降低能源消耗

    2024年03月09日
    浏览(16)
  • 3、TensorFlow教程--- 理解人工智能

    人工智能包括通过机器和特殊计算机系统模拟人类智能的过程。人工智能的示例包括学习、推理和自我校正。人工智能的应用包括语音识别、专家系统、图像识别和机器视觉。 机器学习是人工智能的一个分支,它处理可以学习任何新数据和数据模式的系统和算法。 让我们关

    2024年02月08日
    浏览(16)
  • 人工智能教程(四):概率论入门

    目录 前言 TensorFlow 入门 SymPy 入门 概率论入门 前言 前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。 点击跳转到网站 在本系列的 上一篇文章 中,我们进一步讨论了矩阵和线性代数,并学习了用 JupyterLab 来运行 Python 代码。在本

    2024年02月03日
    浏览(32)
  • 使用人工智能预测日本福岛核污水排放对环境的影响

    1.1 背景介绍 自2011年发生福岛核事故以来,日本一直在努力处理事故造成的影响。在事故后的这段时间里,大量放射性物质被释放到环境中,其中一项重要的决策是如何处理核污水。随着时间的推移,累积的核污水导致了对海洋生态系统和人类健康的不确定风险。为了更好地

    2024年01月16日
    浏览(17)
  • 自主行为与环境适应的挑战:人工智能在能源领域的应用

    能源领域是人工智能(AI)的一个重要应用领域,因为能源系统具有复杂性、不确定性和实时性等特点。在过去的几年里,人工智能技术在能源领域取得了一定的进展,但仍然面临着许多挑战。这篇文章将从以下几个方面进行探讨: 背景介绍 核心概念与联系 核心算法原理和具体

    2024年02月22日
    浏览(17)
  • 人工智能教程(三):更多有用的 Python 库

    目录 前言 推荐 JupyterLab 入门 复杂的矩阵运算 其它人工智能和机器学习的 Python 库 前言 在本系列的上一篇人工智能教程(二):人工智能的历史以及再探矩阵中,我们回顾了人工智能的历史,然后详细地讨论了矩阵。在本系列的第三篇文章中,我们将了解更多的矩阵操作,

    2024年02月05日
    浏览(18)
  • 最新人工智能GPT-4免费简单使用教程

    GPT-4比Chatgpt升级了不少,现在还无法使用OpenAI官网或百度文心一言的小伙伴可以尝试以下方法。 打开 nat.dev 登录(sign-in注册时不用填手机号码) 选择GPT-4模型 然后直接空白处用英语或者中文输入问题 想尝试AI作图的方法如下。 打开网页 https://discord.com/ 点击“Open Discord” 发

    2023年04月25日
    浏览(20)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包