### !!!直接使用pip安装,不要使用conda安装,使用conda安装总是会安装成CPU版本。
1.第一次尝试
之前按照官网安装pytorch的命令进行安装,结果安装出来是cpu版本的,试过https://blog.csdn.net/u013468614/article/details/125910538里面的一些方法,但是使用conda安装出来还是CPU版本。
2.第二次尝试
然后,又尝试了把安装包下载到本地,使用
https://zhuanlan.zhihu.com/p/111430478这个方法,又出现了没有torch的错误,查找到说是可以按照pytorch官网的安装方法再安装一遍,但是不管是pip3还是conda,都会把原来的本地的gpu包进行升级,安装的torch也是CPU版本的,所以还是不行。
3. 终于成功
最后,重新建了一个虚拟环境,使用这个一开始的方法,添加了清华镜像源之后直接使用官网pip命令安装,成功了
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
使用的教程是这个https://zhuanlan.zhihu.com/p/672526561
菜鸟折腾了好久终于好了,记录一下自己的操作过程,如果能对和我一样的朋友有帮助,那也是我的荣幸了。