安装pytorch和torchvison
1、了解:
在安装pytorch前需要安装cuda和cudnn,该教程暂未涉及,后续会添加。
个人认为最好的安装方法是下载wheel轮子手动安装,直接用官方给的命令有时候网络啥的出现问题就安装失败了,而且安装慢。
2、第一步,下载对应的pytorch和torchvision的wheel包
2.1 下载torch的wheel文件
下载时要对应版本,一个是cuda版本一个是python版本。举个例子,我的cuda是11.7的,python是3.8的,如下所示:
注:前面的cu117代表cuda11.7版本的,cp38代表python3.8版本的,win_amd64是windows的,linux_x86是Linux系统的。下面还有几个torch也是cu117和cp38,只是torch本身版本不一样,个人理解没啥大问题,就看你自己想用哪个。
2.2 下载torchvision的wheel文件
下载torchvision和下载torch时差不多,主要在选择torchvision版本时要和torch一致,图片如下:
3、安装wheel文件
这两个包随便放哪都行,只要知道地址。
利用conda activate 进入自己的环境。
利用下面命令进行安装
pip install "torch的地址"
pip install "torchvision的地址"
这是我运行时的命令:
4、验证是否安装成功
利用conda activate进入虚拟环境
输入如下命令检测:
5、总结
以上就是本人安装pytorch中的过程,希望对各位有用。后续安装cuda和cudnn也会整理出来的。