pytorch配置镜像源配置(个人笔记windows)

本文指导如何通过conda创建并管理pytorch环境,包括检查显卡配置、选择合适的CUDA版本,以及在有英伟达显卡和CPU环境下安装PyTorch,同时提及了使用清华镜像源加速下载过程。
摘要由CSDN通过智能技术生成

首先安装pytorch需要先配置conda,可以参考下面这篇文章:

conda配置及安装

安装好后可以搜索到

打开显示如下:

显示base环境,这表明其安装成功。

接下来我们需要尽量避免在base环境中配置文件,我们这里选择创建一个名字为pytorch,python版本为3.6的环境

conda create -n pytorch python=3.6 

接下来弹出提示,按下y即可完成安装

使用 conda info --envs可以查看已创建的环境

其中*代表你现在所处的环境。

使用 activate pytorch 即可转换到pytorch环境

基本工作完成后我们接下来配置pytorch

pytorch安装:

首先我们需要了解自己电脑的配置,我们需要查看自己电脑是否具有英伟达显卡,并且了解其型号。

有英伟达显卡的:

可以在这里查看型号。

接下来需要查看本机cuda版本:

使用命令 nvidia-smi查看:

前两个必须大于400 cuda版本决定你后面的安装pytorch的版本。

接下来进入pytorch官网:

pytorch官网这个可能进去比较慢,毕竟是国外网站,可以使用小梯子。

这里是conda安装,这里如果可以直接安装的话最好,本人就是一直报错才选择使用pip安装,并且使用国内镜像源进行安装。

这里我们直接复制其代码如下:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

接下来我们使用清华镜像源:-i https://pypi.tuna.tsinghua.edu.cn/simple some-package

就是在上面的代码的后面加上这个:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

ok然后直接复制粘贴运行。

无英伟达版本就是选择cpu同理即可。

接下来我们验证一下:

打开python 输入import torch:

(先写到这,发现一些问题,不过上面pytorch确实可以用,等我找到一些问题的解决 方法会更新的)

(已找到解决方法,这两天会在下一篇文章详细说明)

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值