清华园镜像下载pytorch-gpu
下载 PyTorch GPU 版本
为了从清华大学的镜像站点下载支持 GPU 的 PyTorch,可以按照以下方法操作:
对于稳定版本安装,推荐使用 pip
或者 conda
工具来简化依赖管理过程[^1]。
使用 pip 安装
通过命令行执行如下指令完成带有 CUDA 支持的 PyTorch 安装:
pip install torch torchvision torchaudio --extra-index-url https://pypi.tuna.tsinghua.edu.cn/simple
这条命令会自动选择适合当前环境的最佳匹配版本并利用清华源加速下载速度。
使用 conda 安装
如果偏好 Anaconda 环境,则可以通过 Miniconda 或 Anaconda 来安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c nvidia -c defaults
注意这里指定了 -c
参数指向特定渠道以确保获取到由 NVIDIA 提供优化过的二进制文件,并且选择了合适的 CUDA Toolkit 版本号以便兼容目标硬件设备[^2]。
清华园镜像下载pytorch(GPU版本)
清华大学开源软件镜像站提供PyTorch的预构建GPU版本下载。首先,你需要访问清华大学开源软件镜像站的官方网站,通常地址是 https://mirrors.tuna.tsinghua.edu.cn
或类似的URL。
要在该站点找到PyTorch的GPU版本,你可以按照以下步骤操作:
登录网站后,在搜索框中输入“PyTorch”并选择“CUDA版本”(比如PyTorch+cuDNN),因为这表示包含了对NVIDIA GPU的支持。
点击搜索结果进入PyTorch的相关页面,注意查看版本信息,确保选择适合你项目的稳定版或者最新版。
找到对应你Python环境(如Python 3.x)的下载链接,例如
torch-xx.xx-cp3x-none-linux_x86_64.whl
这样的文件。下载完成后,可以在你的Python环境中通过pip安装,例如:
pip install --index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/simple torch torchvision # 如果需要vision模块
anaconda配置pytorch环境gpu清华园
配置 Anaconda 中支持 GPU 的 Pytorch 环境
为了在 Anaconda 中通过清华大学镜像源配置支持 GPU 的 Pytorch 环境,可以遵循以下方法:
设置清华镜像源
首先,在终端或者命令提示符中运行以下命令来设置清华镜像作为默认的 Conda 渠道:
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 --set show_channel_urls yes
这些命令将添加清华大学开源软件镜像站点到 Conda 的渠道列表,并启用显示渠道 URL 功能以便确认安装包来自该镜像[^2]。
创建新的 Conda 环境并激活它
创建一个新的 Conda 环境用于安装 Pytorch 和其他依赖项。假设我们将其命名为 gpu_env
:
conda create -n gpu_env python=3.8
接着激活这个新环境:
conda activate gpu_env
安装 CUDA 支持的 Pytorch 版本
根据目标硬件的支持情况选择合适的 CUDA 版本号(例如 cuda102 对应于 CUDA 10.2)。以下是针对不同 CUDA 版本的具体安装命令示例: 对于 CUDA 10.2 用户可执行如下操作:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch -c nvidia
上述命令中的 -c pytorch
表明从官方 PyTorch 渠道获取资源;而 -c nvidia
则是从 NVIDIA 提供的内容里下载必要的驱动程序组件[^3]。
添加 Jupyter 内核至新建环境中
完成以上步骤之后,为了让 Jupyter Notebook 或 Lab 能够识别此特定版本下的 Pytorch 库文件,则需把当前活动状态所对应的虚拟环境加入 notebook kernel 当中:
python -m ipykernel install --user --name=gpu_env --display-name "Python (GPU)"
这一步骤使得可以在启动后的交互界面看到名为 “Python (GPU)” 的选项卡可供切换使用[^1]。
测试安装是否成功
最后验证一下整个流程下来的结果是不是符合预期效果——即能否正常调用带有 GPU 加速特性的 PyTorch 函数库。打开 Python 解释器输入下面几行测试代码看看是否有错误发生以及设备名称是否指向了正确的显卡型号。
import torch
print(torch.cuda.is_available()) # 输出 True 如果有可用的CUDA 设备
print(torch.version.cuda) # 显示已加载的 CUDA 版本字符串
print(torch.cuda.get_device_name(0)) # 获取第一个 GPU 卡片的名字
如果一切顺利的话,你应该能够看到关于你的图形处理器的相关信息被打印出来。
相关推荐














