总结一下搭建vscode+anaconda深度学习环境的过程,安装pytorch的过程,我认为最简单并且最不容易出错的安装方式如下,我用的是vscode+Anaconda+Pytorch
1、下载对应的cuda
首先要确保已经配置好了vscode和anaconda3.0的环境,为了保证之后再vscode中能够成功import torch包,我们需要先根据自己的nvidia版本下载对应的cuda。
(关于cuda,pytorch的关系,以及如何找到对应的cuda,这个链接说的很明白可以看看有助于对版本的理解:https://www.jianshu.com/p/ac70300b598b
)
关于cuda的安装我按照这个教程走的https://www.cnblogs.com/Renyi-Fan/p/13456984.html
2、安装Pytorch
安装好cuda之后,就是安装Pytorch,按照这个教程绝对不会踩坑!(前提是确保自己的版本都是对的,我的nvidia为11.几,python为3.8,cuda为10.1,torch和torchversion包下载的都是和这个博客里一样的,最后没有出现问题)https://blog.csdn.net/qq_36468195/article/details/108706428
但是在下图所示的步骤中,也就是安装已经下载好的whl时,报了错错误:(样子是下载到一半,跳一大片红)
socket.timeout: The read operation timed out
解决办法:pip指令修改成pip --default-timeout=100 install 包名
(不用加别的镜像源什么的)
3、Vscode中import torch包
这个需要注意,因为按照上面那个不会踩坑的教程中,我们是在Anaconda里建设了一个pytorch的虚拟环境,用人话说就是,你之前在Anaconda prompt里下载的比如matplotlib包这种,在这个环境里都没有。
所以在Anaconda prompt中,先输入以下指令,激活虚拟环境activate pytorch-gpu
此时,用pip安装torch包或者matplotlib包,就可以了。
下载完之后,记得在vscode里,更换pyhton选择器:
1、Ctrl + shift + P
2、输入:Python:Select Interpreter
3、选择PyTorch解释器
这个时候在代码中import torch包应该就没有问题了!
接下来就是最近在做的风格迁移相关的一些小问题,比如超时问题,图片路径什么的,具体问题具体解决吧!最后还是成功跑出来了~