Vscode+Anaconda安装Cuda+Pytorch搭建风格迁移相关的深度学习环境

总结一下搭建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时,报了错Vscode+Anaconda安装Cuda+Pytorch搭建风格迁移相关的深度学习环境_第1张图片错误:(样子是下载到一半,跳一大片红)

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解释器

Vscode+Anaconda安装Cuda+Pytorch搭建风格迁移相关的深度学习环境_第2张图片

这个时候在代码中import torch包应该就没有问题了!
接下来就是最近在做的风格迁移相关的一些小问题,比如超时问题,图片路径什么的,具体问题具体解决吧!最后还是成功跑出来了~
Vscode+Anaconda安装Cuda+Pytorch搭建风格迁移相关的深度学习环境_第3张图片

你可能感兴趣的:(深度学习,python,深度学习,pytorch)