win10安装pytorch

简介: win10安装pytorch

1. 创建和激活新的环境


conda create -n pytorch python=3.6
conda activate pytorch


2. 安装pytorch前的准备


2.1 查看设备型号


安装pytorch之前需要安装cuda,调用gpu加速,所以需要知道gpu的型号


1、查看显卡型号


方法一:打开设备管理器->显示配适器,查询到为NVIDIA GeForce GTX 1050



方法二:打开任务管理器->性能



2、查看驱动版本


打开NVIDIA控制面板->帮助->系统信息



查看cuda适配版本



根据驱动版本选择对应的cuda版本



2.3 安装cuda


cuda官网链接,翻进去傻瓜式安装


老版本的cuda链接:链接



输入nvcc -V查看是否安装成功



3. 安装pytorch


4、根据步骤1、2查询到的信息,进入pytorch官网,https://pytorch.org/,选择对应版本,复制指令,进入anaconda对应环境下输入指令安装pytorch


先换源,下载的快


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


后复制下面指令,注意去掉-c pytorch


conda install pytorch torchvision torchaudio cudatoolkit=10.2



报错:



看别人博客链接,说是因为anaconda不是默认装在C盘才报错


解决方法步骤:找到anaconda文件夹右击打开属性->安全->修改为system



再一次运行安装指令,之后便安装成功。



4. 测试


打开pycharm或者就在终端上测试:


import torch
torch.cuda.is_available()


返回为True



相关文章
|
1月前
|
并行计算 PyTorch 算法框架/工具
anaconda+pytorch+opencv安装及环境配置
anaconda+pytorch+opencv安装及环境配置
106 0
|
8月前
|
机器学习/深度学习 PyTorch 算法框架/工具
Pytorch学习笔记(1):基本概念、安装、张量操作、逻辑回归
Pytorch学习笔记(1):基本概念、安装、张量操作、逻辑回归
148 0
Pytorch学习笔记(1):基本概念、安装、张量操作、逻辑回归
|
3月前
|
机器学习/深度学习 人工智能 PyTorch
【Hello AI】安装和使用AIACC-AGSpeed(优化PyTorch深度学习模型)
AIACC-AGSpeed(简称AGSpeed)专注于优化PyTorch深度学习模型在阿里云GPU异构计算实例上的计算性能,相比原始的神龙AI加速引擎AIACC,可以实现无感的计算优化性能。本文为您介绍安装和使用AGSpeed的方法。
|
4月前
|
机器学习/深度学习 并行计算 PyTorch
win10安装minconda+pytorch
本文主要说明在win10笔记本上的安装minconda和pytorch,以图文结合的方式记录安装步骤,供参考学习。
63 1
|
5月前
|
数据可视化 PyTorch TensorFlow
Keras和Tensorflow(CPU)安装、Pytorch(CPU和GPU)安装以及jupyter使用虚拟环境
Keras和Tensorflow(CPU)安装、Pytorch(CPU和GPU)安装以及jupyter使用虚拟环境
97 0
|
6月前
|
并行计算 PyTorch Linux
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统下GPU版PyTorch(CUDA 12.1)快速安装
2589 0
|
6月前
|
并行计算 PyTorch Linux
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统快速安装指南
pytorch安装GPU版本 (Cuda12.1)教程: Windows、Mac和Linux系统快速安装指南
1302 0
|
8月前
|
并行计算 PyTorch 算法框架/工具
Anaconda 安装并使用 PyTorch(PyCharm)
Anaconda 安装并使用 PyTorch(PyCharm)
461 1
|
8月前
|
并行计算 PyTorch 算法框架/工具
Pytorch安装和检验cuda和cudnn是否可用
Pytorch安装和检验cuda和cudnn是否可用
553 0
|
9月前
|
机器学习/深度学习 人工智能 Linux
Linux和Windows系统下安装深度学习框架所需支持:Anaconda、Paddlepaddle、Paddlenlp、pytorch,含GPU、CPU版本详细安装过程
Linux和Windows系统下安装深度学习框架所需支持:Anaconda、Paddlepaddle、Paddlenlp、pytorch,含GPU、CPU版本详细安装过程
Linux和Windows系统下安装深度学习框架所需支持:Anaconda、Paddlepaddle、Paddlenlp、pytorch,含GPU、CPU版本详细安装过程