pytorch+GPU 清华镜像源安装(anaconda)

本文详细介绍了如何在Anaconda环境下配置PyTorch GPU版本,包括更换清华镜像源、安装PyTorch及核心模块测试。适用于希望快速上手深度学习GPU加速的开发者。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

0. 把所有事情都完成之后

本人是把所有的内容都放到了一个base(root)里面,因此所有内容都在base里面的终端中输入。如果我们想要使用这个环境,只需要 conda activate base(root) 就行。如果是在pycharm的环境里面,只需要setting一下这个环境即可;如果是在vscode里面,仅仅需要调整左下角的环境即可。

1. 换清华镜像源

打开终端,并输入:

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/cloud/pytorch/
conda config --set show_channel_urls yes  

(其实更改的是C盘中的一个文件)

2. 安装pytorch+gpu

在这里插入图片描述
在pytorch官网( https://pytorch.org/ )上可以很轻松地找到:我们需要的内容。但是同时,记住我们需要进行一定的更改,把最后的 -c pytorch去掉。例如:

conda install pytorch torchvision cudatoolkit=10.2

3. 等待安装即可

4. 几个核心模块的测评:

pytorch

import torch
x = torch.empty(3,5)
print(x)

看能不能出来一个3*5的矩阵。

GPU

print(torch.cuda.is_available())  

如果结果是True即可。

评论 14
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值