pytorch安装
Posted 今天也要debug
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch安装相关的知识,希望对你有一定的参考价值。
pytorch版本需要和cuda版本相适应,所以首先查看自己电脑的cuda版本
(10条消息) CUDA学习(一)——如何查看自己CUDA版本?_溪午不闻钟_的博客-CSDN博客_cuda version
通过以下命令行安装对应版本的pytorch
pip --default-timeout=1000 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
以上是关于pytorch安装的主要内容,如果未能解决你的问题,请参考以下文章
《动手学深度学习》线性回归的简洁实现(linear-regression-pytorch)
《动手学深度学习》线性回归的简洁实现(linear-regression-pytorch)
PyTorch深度学习实战-1-Python3环境与PyTorch安装
Pytorch深度学习实战3-6:详解网络骨架模块nn.Module(附实例)