pytorch的安装—pip

之前我有写过使用anaconda安装pytorch的文章,但是当时出现了无法解决的问题,安装的pytorch不管如何,都是下载的cpu版本的,于是我重复很多次,检查了cuda版本,cudaa版本,都是一一对应的,但是用conda下载就是不行,直到我看到了这样一篇文章:

安装GPU版本的pytorch(解决pytorch安装时默认安装CPU版本的问题)保姆级教程

这篇文章给出了他的解决方案,但是对我来说,还是不行,可是我在这篇文章里又看到了这样一条评论

因此我决定,使用pip安装pytorch

最后发现,我只需要一个代码直接就下载下来了......

但是如果我直接下载的话,还是会需要进行包的调用,见我的上一篇文章,《在anaconda下安装pytorch的步骤》的后半部分,还有《关于包的调用》,这里我们直接在pip安装的时候就指定好安装路径:在《关于包的调用》里面有写到:

这里我所安装的torch版本是1.10.1,cuda版本是10.2,写入的代码如下:

pip install --target=D:\app\anaconda3\envs\mmgen\Lib\site-packages torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1+cu102 -f [https://download.pytorch.org/whl/cu102/torch_stable.html](https://download.pytorch.org/whl/cu102/torch_stable.html)

结果可以正常调用!!!

posted @ 2023-05-16 16:04  top王  阅读(228)  评论(0编辑  收藏  举报