栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Pytorch安装教程

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Pytorch安装教程

Pytorch安装教程(使用版本CUDA == 10.2)

前言

1.安装Anaconda

2.查看CUDA版本

3.安装Pytorch

4.测试是否安装成功


前言

       在测试中一直提示False,后来又看了下其他教程,发现其他博主都是针对CUDA>10.2版本进行安装,后来参考了好多文章,终于找到如何解决这个问题,现在就把整体的安装教程写出来供大家参考。

安装Anaconda

理论知识就不给大家整理了,直接开始安装教程!!!

Anaconda下载链接:

官方下载链接:Anaconda | The World's Most Popular Data Science Platform

可能下载速度比较慢,下面给出清华大学开源镜像下载地址

清华源下载地址:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

        往下翻,在清华大学开源软件镜像站中选取较新的的Anaconda安装包

下载完成后双击,开始安装Anaconda

 点击I Agree ,同意安装协议

 

 选择Just Me,点击Next进行下一步

 选择安装目录,点击Next进行下一步

 勾选下面两个选项,进行安装

2.查看CUDA版本

3.安装Pytorch

创建pytorch环境,PyTorch为环境名

conda create -n PyTorch python=3.8
查看已创建完成的环境
conda info --envs
进入到环境名为PyTorch的环境中
conda activate PyTorch
开始安装pytorch环境
pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
4.测试是否安装成功
python

import torch

print(torch.__version__) # pytorch版本

print(torch.version.cuda) # cuda版本

print(torch.cuda.is_available()) # 查看cuda是否可用

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1009553.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号