pip安装pytorch清华_win10使用清华源快速安装pytorch-GPU版(推荐)

检查自己的cuda是否安装好

在anaconda prompt中输入

nvcc -V

显示如上面表示安装好了。

配置清华园下载环境

同样在在anaconda prompt中输入

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda config --set show_channel_urls yes

然后删除:C:\Users\Administrator 目录下的 .condarc 文件,打开文件删除defaults那一行。如下图所示:

使用清华源安装PyTorch以及对应版本的CUDA,您可以按照以下步骤进行: 1. 首先,您需要确定您的系统环境,包括操作系统版本、Python版本以及CUDA版本(如果您打算安装GPU版本PyTorch)。对于CUDA,通常需要与PyTorch兼容的版本。 2. 访问PyTorch官方网站的安装指南页面(https://pytorch.org/get-started/locally/),根据您的环境选择合适的PyTorch版本。在该页面上,您可以选择使用Pip”作为包管理工具,并勾选使用中国大陆的清华源。 3. 确定选项后,网站会提供一条命令,该命令包含了使用清华源安装特定PyTorch版本的指令。复制这条命令。 4. 打开您的终端或命令行界面,粘贴复制的命令并执行。例如,如果命令是这样的: ``` pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html ``` 这将安装与您的CUDA兼容的最新稳定版本PyTorch(CPU版本)。 5. 如果您需要安装GPU版本PyTorch,您需要找到与您的CUDA版本对应的pytorchtorchvision版本。您可以在PyTorch官方网站或者清华大学开源软件镜像站找到相应的whl文件链接。安装命令类似于: ``` pip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 注意替换`1.8.0+cu101`和`0.9.0+cu101`为您的CUDA版本和所需的PyTorch版本。 请注意,上述命令和步骤仅适用于从PyTorch 1.8.0版本开始,对于旧版本PyTorch安装命令可能有所不同,请参考对应版本安装指南。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值