关于pytorch的安装

1.如果系统没有显卡,可以使用cpu进行系统的训练
本次使用命令进行安装

pip3 install torch==2.0.0+cpu torchvision==0.2.2 -f https://download.pytorch.org/whl/torch_stable.html

如果该命令执行错误,可以参照错误的命令更改相应的版本号。
2.我的cuda版本为11.1,为此我安装了11.1cuda对应的pytorch,命令行代码如下:

pip torch-1.9.1+cu111-cp39-cp39-win_amd64.whl

cu代表你自己的cuda版本,一般来说就是要低于或者是等于自己安装的cuda版本号。
torch-1.7.1+cu110-cp39-cp39-win_amd64.whl这个文件是我自己下载好的,因为之前使用官网给的命令行安装老是出错,所以自己按照网址(这个网址就是官网命令行中的网址,自己可以找找),找了whl自己下载了,下载之后命令行安装出错错误截图
说明这个命令不适合在此环境下运行,作者出现这个错误的原因是作者的python版本号为3.11,但是命令行中是3.9版本。这里我建议大家更改一下自己的python环境,尽量不要去修改文件名字达到成功安装的目的,最终就安装成功了,希望大家一切顺利。cuda安装过程遇到麻烦也是比较头疼,大家可以找资源,或者在我评论区下面留言,希望本篇博客可以为您带来帮助。

你可能感兴趣的:(pytorch,深度学习,python)