ubuntu anaconda安装pytorch

最新推荐文章于 2023-11-09 09:40:04 发布

原创最新推荐文章于 2023-11-09 09:40:04 发布 · 832 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#ubuntu装pytorch

学习心得同时被 2 个专栏收录

28 篇文章

订阅专栏

笔记

17 篇文章

订阅专栏

本文介绍如何配置清华大学的Anaconda镜像源，以加速Python包的下载，并提供针对不同CUDA版本的PyTorch及torchvision的安装命令。通过修改.condarc文件，可以有效提高conda环境下软件包的安装速度。

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

清华conda源地址https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/，在里面可以查看pytorch和python版本的组合是否有资源
.condarc文件内容：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2

  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/simpleitk/
show_channel_urls: true

安装命令：https://pytorch.org/get-started/previous-versions/
比如：

# CUDA 10.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 

# CUDA 9.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda90 

# CUDA 8.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 

# CPU Only
conda install pytorch-cpu==1.0.0 torchvision-cpu==0.2.1 cpuonly