正常来说,gpu版本的torch,版本后面都是带有cuda的版本号的。
但是,我在使用anaconda 下载pytorch框架的时候,发现,他给我整出来的是cpu。
原因大致上是因为使用了清华源。
建议看看下面两篇文章。
https://blog.csdn.net/Andy_Luke/article/details/122503884
https://blog.csdn.net/qq_46126258/article/details/112708781
可以看到,gpu版本的,是带有cuda版本的。
https://pytorch.org/get-started/locally/#windows-pip
可以直接复制这里面的命令,进行下载,但是会很慢很慢。
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
不过,我听说,如果你之前下载过gpu版本的,可以直接将虚拟环境中与pytorch相关的cpu版本的删掉,复制gpu版本的,也是ok的。
比如我这些是gpu版本的,我试着将他复制到conda创建的虚拟环境中,发现还真的可行。
就免去了,重新下载。