安装pytorch

官方网址：

Start Locally | PyTorchhttps://pytorch.org/get-started/locally/

官网的conda安装方式，没有指定各种组件的版本，因为好多算法是规定了pytorch版本，这种可读性不高的命令行安装方式我没有采取。

我采取的是找到之前的版本，但是要求他比较新，因为算法在迭代，pytorch版本低，之后还得重新配置虚拟环境。

选一个合适的版本，根据我之前的学习现在比较新的算法都需要pytorch1.10以上了，我选择1.11.0。

注意！！！pytorch1.6.0在练习pytorch时会出现torch.nn没有代码提示！！！避免安装pytorch1.6.0.

以下是详细的配置环境过程

采用conda安装方式,名称（-n）为pytorch，python版本为3.7。
conda create -n pytorch python=3.7
进入虚拟环境。
conda activate pytorch
采用pip安装，第一步就安装了pip安装工具
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113
查看cuda版本，更新的最新肯定是没问题的，向下兼容。
nvidia-smi

注意cuda版本是向下兼容的，把显卡驱动升级到最新即可。

检查pytorch安装是否成功

以下代码全部运行成功，才算安装成功。

python
import torch
import torchvision
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())
exit()

总结

1.注意别安pytorch1.6.0版本，torch.nn没有代码提示，不利于学习。

2.cuda版本不用严格与系统的cuda版本一致，但是系统的cuda版本要比pytorch要求的cuda版本高。

1.pytorch学习:安装pytorch相关推荐

Pytorch学习 - Task5 PyTorch卷积层原理和使用
Pytorch学习 - Task5 PyTorch卷积层原理和使用 1. 卷积层 (1)介绍 (torch.nn下的) 1) class torch.nn.Conv1d() 一维卷积层 2) clas ...
Pytorch学习 - Task6 PyTorch常见的损失函数和优化器使用
Pytorch学习 - Task6 PyTorch常见的损失函数和优化器使用官方参考链接 1. 损失函数 (1)BCELoss 二分类计算公式小例子: (2) BCEWithLogitsLoss ...
PyTorch学习记录——PyTorch进阶训练技巧
PyTorch学习记录--PyTorch进阶训练技巧 1.自定义损失函数 1.1 以函数的方式定义损失函数 1.2 以类的方式定义损失函数 1.3 比较与思考 2.动态调整学习率 2.1 官方提供的s ...
PyTorch学习记录——PyTorch生态
Pytorch的强大并不仅局限于自身的易用性,更在于开源社区围绕PyTorch所产生的一系列工具包(一般是Python package)和程序,这些优秀的工具包极大地方便了PyTorch在特定领域的使 ...
PyTorch学习笔记——pytorch图像处理（transforms）
原始图像 2.图像处理.转不同格式显示 import torch import torchvision import torchvision.transforms as transforms impo ...
深度学习之pytorch(一) 环境安装
快2周没更新博客了,别问为什么!我才不会告诉你忙的飞起!说心里话,个人更喜欢用tensorflow来做深度学习! 一.环境配置 1.win10 64位系统: 2.anaconda3: 3.不管是CPU ...
PyTorch学习笔记（一）：PyTorch环境安装
往期学习资料推荐: 1.Pytorch实战笔记_GoAI的博客-CSDN博客 2.Pytorch入门教程_GoAI的博客-CSDN博客本系列目录: PyTorch学习笔记(一):PyTorch环境安 ...
Pytorch学习笔记总结
往期Pytorch学习笔记总结: 1.Pytorch实战笔记_GoAI的博客-CSDN博客 2.Pytorch入门教程_GoAI的博客-CSDN博客 Pytorch系列目录: PyTorch学习笔记( ...
PyTorch学习笔记（七）：PyTorch可视化
PyTorch可视化往期学习资料推荐: 1.Pytorch实战笔记_GoAI的博客-CSDN博客 2.Pytorch入门教程_GoAI的博客-CSDN博客本系列目录: PyTorch学习笔记(一) ...

1.pytorch学习:安装pytorch

安装pytorch

检查pytorch安装是否成功

总结

1.pytorch学习:安装pytorch相关推荐

最新文章

热门文章