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时,报了错在这里插入图片描述错误:(样子是下载到一半,跳一大片红)

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包应该就没有问题了!
接下来就是最近在做的风格迁移相关的一些小问题,比如超时问题,图片路径什么的,具体问题具体解决吧!最后还是成功跑出来了~
在这里插入图片描述

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值