如何正确配置pytorch环境

1 各种镜像源的问题

因为pytorch原生的服务器在国外,而pytorch(cuda版本)大小为1G以上,使用国外源下载是不太可能的。因此衍生出了很多国内的下载源。pytorch主要包括三个包,pytorch、torchversion和torchaudio,另外在加一个cudatoolkit。

清华源:

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

南京大学源:

https://mirrors.nju.edu.cn/anaconda/cloud/pytorch/win-64/

源的概念就是一个下载地址,在这个地址下,有需要的pytorch安装包,在此地址下就可以快速的安装好pytorch的环境。

2 下载源的切换

b站很多博主都是直接把一些源直接添加到anaconda的资源库里面,这样会造成一些混乱。因为不同的源里的资源是不同的,有的是用来下载pytorch的,可能还有一些是用来下载其他的必要的安装包的。因此在进行下载时,手动配置下载地址即可,默认下载地址保持不变。

配置下载地址的方法也很简单:

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

只需要在-c 后手动把下载源的链接加上面就行了。

3 下载指令

下载指令有conda和pip,这两个都是下载的方式,有的使用conda,而有时候使用pip。这两个命令相当于从不同的软件商店下载同一个软件的关系。

4 本地的下载

也可以把pytorch下载到本地进行安装,下载文件格式为.whl

conda install --use-local dir
pip install dir

4 一些指令

conda list #用来查看虚拟环境中的安装包,各种插件
conda install #安装需要的安装包
conda uninstall #卸载需要的安装包
conda config --show #下载通道查看
conda config --get  #下载通道查看
conda config --add channels 下载源链接  #添加通道
conda config --remove channels 链接     #移除通道
conda install nb_conda #安装jupyter
jupyter notebook #用来启动安装好的jupyter
conda/pip install opencv-python #opencv的安装

5 pytorch准确安装检验

python #进入python环境
import torch #进入torch
torch.cuda.is_available() #验证是否正常使用
true

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值