Anaconda3+PyTorch安装教程

简介: Anaconda3+PyTorch安装教程

一、安装前准备


电脑需要有NVIDA的独立显卡;

电脑要有NVIDA显卡驱动程序(NVIDA控制面板)

微软官网下载安装 Microsoft Visual Studio(申请一个微软账号)

官网下载安装 CUDA Toolkit

检验CUDA安装是否成功

CUDA安装成功后的使用


二、前提


CUDA在安装之前,需要看NVIDA的官方CUDA安装文档。


CUDA 11.2: https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html


下载安装CUDA需要有NVIDA的显卡、Windows系统、Visual Studio。


三、查看电脑显卡驱动信息


一般新买的电脑都带有。


官网下载:https://www.nvidia.cn/geforce/drivers/


四、下载安装 Visual Studio(非必要)


官网:https://visualstudio.microsoft.com/zh-hans/(选择社区版即可)


勾选“C++的桌面开发“,然后下载到本地。


选择安装位置进行安装。


021fc5cbeded56c5fa34f67b0b65f0da.png


五、NVIDIA CUDA Toolkit 10.2安装


cuda只能运行在英伟达的显卡上

83eda7556354e2bfaca9187847fb654b.png


第一步,系统检查,如果下载的和NVIDIA控制面板结果一致,则显示绿色通过提示;


第二步,许可协议,同意并继续;


第三步,一般选择自定义安装,选择全部即可


3e1ae82a65ada07bd88e49636a040a92.png


第四步,选择安装位置,可以自己设置,也可以遵循系统设置(装在C盘)


第五步,如果没有安装Visual Studio,需要勾选下面方框,然后Next。


第六步,查看版本是否安装成功。



nvcc -V                            # 出现下列说明安装成功


nvcc: NVIDIA ® Cuda compiler driver

Copyright © 2005-2019 NVIDIA Corporation

Buiit on Wed Oct 23 19:32:27 Pacific_Daylight_Time 2019

Cuda compilation tools, release 10. 2, v10.2. 89


如果安装不成功,则需要配置环境变量。


电脑右键属性—>高级系统设置—>环境变量(系统变量path)


50af49c4000e5c2e821145ebb87c0acc.png


六、创建PyTorch环境


6.1 方法一:


卸载原来的pytorch和torchvision,之前没有安装则忽略这一步。


打开Anaconda Prompt:


①创建PyTorch环境


conda create -n pytorch python=3.7


②激活PyTorch环境


conda activate pytorch            # pytorch为环境名


③登录PyTorch:https://pytorch.org/,找到命令:


(pytorch)C:\user>conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch


④查看是否安装成功:


nvcc -V


⑤退出:


deactivate pytorch


⑥删除虚拟环境


conda remove -n 环境名称 --all


6.2 方法二(推荐)


直接以管理员身份运行cmd,然后在串口输入:


conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch


cfdf25224679e443e75f793b0e56796a.png


七、pycharm中添加conda环境(不选择创建虚拟环境)


在pycharm中新建一个工程(new project)—>选择已经存在的环境解释器(Existing interpreter)—>选择(Conda Environment)—>选择Anaconda安装的python解释器(python.exe文件)—>勾选Make available to all projects。


说明:这种情况方法简单,但是会造成环境污染。自己电脑中包会非常多,不便于后期管理。


7bf1165bbcc2f3c93f4e7774589c7f57.png


八、测试pytorch是否安装成功


import torch
print(torch.__version__)    # 打印pytorch的版本
print("gpu:",torch.cuda.is_available())   # 查看torch是否支持gpu


E:\A_code\1\venv\Scripts\python.exe E:/A_code/1/test.py


1.8.1

gpu: True

Process finished with exit code 0


说明环境已经配置成功


九、当程序运行时,在windows终端中查看当前GPU的使用率


nvidia-smi
相关实践学习
部署Stable Diffusion玩转AI绘画(GPU云服务器)
本实验通过在ECS上从零开始部署Stable Diffusion来进行AI绘画创作,开启AIGC盲盒。
目录
相关文章
|
2月前
|
存储 物联网 PyTorch
基于PyTorch的大语言模型微调指南:Torchtune完整教程与代码示例
**Torchtune**是由PyTorch团队开发的一个专门用于LLM微调的库。它旨在简化LLM的微调流程,提供了一系列高级API和预置的最佳实践
234 59
基于PyTorch的大语言模型微调指南:Torchtune完整教程与代码示例
|
3月前
|
PyTorch Linux 算法框架/工具
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
这篇文章是关于如何使用Anaconda进行Python环境管理,包括下载、安装、配置环境变量、创建多版本Python环境、安装PyTorch以及使用Jupyter Notebook的详细指南。
432 1
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
|
3月前
|
并行计算 PyTorch TensorFlow
Ubuntu安装笔记(一):安装显卡驱动、cuda/cudnn、Anaconda、Pytorch、Tensorflow、Opencv、Visdom、FFMPEG、卸载一些不必要的预装软件
这篇文章是关于如何在Ubuntu操作系统上安装显卡驱动、CUDA、CUDNN、Anaconda、PyTorch、TensorFlow、OpenCV、FFMPEG以及卸载不必要的预装软件的详细指南。
5990 3
|
2月前
|
并行计算 监控 搜索推荐
使用 PyTorch-BigGraph 构建和部署大规模图嵌入的完整教程
当处理大规模图数据时,复杂性难以避免。PyTorch-BigGraph (PBG) 是一款专为此设计的工具,能够高效处理数十亿节点和边的图数据。PBG通过多GPU或节点无缝扩展,利用高效的分区技术,生成准确的嵌入表示,适用于社交网络、推荐系统和知识图谱等领域。本文详细介绍PBG的设置、训练和优化方法,涵盖环境配置、数据准备、模型训练、性能优化和实际应用案例,帮助读者高效处理大规模图数据。
72 5
|
7月前
|
PyTorch 算法框架/工具
win10下安装pytorch,torchvision遇到的bug
win10下安装pytorch,torchvision遇到的bug
|
3月前
|
PyTorch TensorFlow 算法框架/工具
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
本文提供了在Ubuntu 18.04操作系统的NVIDIA Jetson平台上安装深度学习和计算机视觉相关库的详细步骤,包括PyTorch、OpenCV、ONNX、TensorFlow等。
206 1
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
|
3月前
|
机器学习/深度学习 缓存 PyTorch
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
这篇文章是关于如何下载、安装和配置Miniconda,以及如何使用Miniconda创建和管理Python环境的详细指南。
738 0
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
|
3月前
|
并行计算 PyTorch 编译器
|
3月前
|
并行计算 PyTorch TensorFlow
环境安装(一):Anaconda3+pytorch1.6.0+cuda10.0+cudnn7.6.4+tensorflow1.15+pycocotools+pydensecrf
这篇文章详细介绍了如何在Anaconda环境下安装和配置深度学习所需的库和工具,包括PyTorch 1.6.0、CUDA 10.0、cuDNN 7.6.4、TensorFlow 1.15、pycocotools和pydensecrf,并提供了pip国内镜像源信息以及Jupyter Notebook和Anaconda的基本操作。
382 0
环境安装(一):Anaconda3+pytorch1.6.0+cuda10.0+cudnn7.6.4+tensorflow1.15+pycocotools+pydensecrf
|
5月前
|
并行计算 Ubuntu PyTorch
Ubuntu下CUDA、Conda、Pytorch联合教程
本文是一份Ubuntu系统下安装和配置CUDA、Conda和Pytorch的教程,涵盖了查看显卡驱动、下载安装CUDA、添加环境变量、卸载CUDA、Anaconda的下载安装、环境管理以及Pytorch的安装和验证等步骤。
986 1
Ubuntu下CUDA、Conda、Pytorch联合教程