2021-06-23

小帅321

于 2021-06-23 08:23:20 发布

阅读量146

点赞数 1

分类专栏：学习笔记文章标签： anaconda pytorch gpu

本文链接：https://blog.csdn.net/hlb941029/article/details/118141082

版权

学习笔记专栏收录该内容

12 篇文章 0 订阅

订阅专栏

通过Anaconda轻松安装带CUDA支持的PyTorch，只需几步：1) 安装Anaconda并添加中科大镜像源；2) 在Anaconda Prompt中激活环境，安装CUDA工具包和cuDNN；3) 从PyTorch官网使用pip下载对应版本。

摘要由CSDN通过智能技术生成

安装GPU版本pytorch过程：
发现网上各种安装以及配置cuda环境，总结一种简单方法：
1、安装anaconda（注意安装的时候添加path路径打上对号）
2、打开anaconda Prompt，进入自己的环境，添加中科大镜像源：
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --set show_channel_urls yes
查看镜像源：conda config --show-sources
删除安装源：conda config --remove channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

3、打开anaconda Prompt，进入自己的环境，输入conda install cudatoolkit 下载完成后再输入conda install cudnn

4、进入Pytorch官网，选择下载链接（建议使用pip下载）在这里插入图片描述