【Pytorch】笔记内容007:进行PyTorch1.7.1在Conda虚拟环境进行搭建(GPU版本)

背景

Pytorch作为现在深度学习领域主流的框架之一,恰逢毕设中模型使用Pytorch进行搭建,赶上现在不想做毕设,就来记载下之前我电脑中的Torch1.7虚拟环境配置,啊哈哈哈
补充Tensorflow_GPU安装:https://blog.csdn.net/qq_46343832/article/details/121542099

环境说明

  • Windows10_64位
  • Torch版本:version1.7
  • cuda10.1(原因:安装的为GPU版本,具体选择Cuda可以查看我前面文章Tensorflow中有说明)

具体安装步骤

  1. 使用Conda创建存放Torch的虚拟环境,并进入:
# 删除虚拟环境
conda remove -n your_env_name(虚拟环境名称) --all
# 创建虚拟环境,采用3.8环境
conda create -n torch1.7 python=3.8
# 查看虚拟环境命令
conda env list
# 进入虚拟环境命令
conda activate torch1.7

在这里插入图片描述
2. 进入Pytorch官网,找到Pytorch1.7的GPU版本,注意如果你电脑比较牛,显卡驱动支持,也只直接装最新的版本,步骤都一样:
在这里插入图片描述
3. 找到Tourch1.7对应的Conda命令复制到cmd中的虚拟环境中进行安装,可能安装过程比较缓慢,如果你有时间等下去,可直接等它安装好就可使用,如果中间部分第三方库丢失,可重新执行该命令。如果你不想等,请继续往下看,执行另一种安装方式。
在这里插入图片描述
4. 如果你也出想像我这情况,肯定不想等待吧,哈哈哈哈哈!!!那下面我们开始另外一种安装方式
在这里插入图片描述
① 使用conda安装下cudatoolkit=10.1,cuda选择只要比你电脑驱动支持的cuda版本低就没的问题(查看自己电脑的cuda版本支持看文章开头里说的Tensorflow_GPU安装里有说明)

conda install cudatoolkit=10.1

如果安装还是比较慢!建议换下安装源,比如清华源等
在这里插入图片描述
在这里插入图片描述
② 手动下载torch1.7.1的压缩包.whl文件,再进行pip的安装,比如,我们找到官网提供的torch1.7的pip安装如下:

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

然后我们其后半部分https://download.pytorch.org/whl/torch_stable.html进入官方存放torch压缩文件的位置
③ 搜索torch-1.7.1%2Bcu101找到https://download.pytorch.org/whl/cu101/torch-1.7.1%2Bcu101-cp38-cp38-win_amd64.whl,鼠标右击复制链接地址,采用迅雷下载,为啥用迅雷呢?懂的都懂!!!!
在这里插入图片描述
④ cmd进入下载好的文件所在目录中
在这里插入图片描述
⑤ 使用命令pip install "torch-1.7.1+cu101-cp38-cp38-win_amd64.whl" -i https://pypi.douban.com/simple/进行安装

⑥ 对pip安装中剩余的第三方库进行安装

pip install torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/

在这里插入图片描述

测试安装成功

在这里插入图片描述

  • 5
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Chuckie今天也要学习!

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值