pytorch1.2安装（Anaconda、cuda10.0、cudnn7.5）

最新推荐文章于 2024-09-10 23:52:11 发布

木子爱学习

最新推荐文章于 2024-09-10 23:52:11 发布

阅读量1.0k

点赞数

CC 4.0 BY-SA版权

文章标签： pytorch 深度学习神经网络

本文链接：https://blog.youkuaiyun.com/qq_40467402/article/details/107991054

本文详细介绍了如何使用conda在虚拟环境中创建并安装PyTorch 1.2.0版本，特别针对CUDA 10.0环境。包括设置清华镜像源、安装所需依赖及PyTorch的具体步骤。

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

首先需要新建虚拟环境

conda create –n your_name python=3.6
//这里的yourname是虚拟环境的名字，取什么都可以

激活环境

activate  your_name

添加清华镜像源

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/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes

安装代码

去官网找安装代码 https://pytorch.org/get-started/previous-versions/
在这里插入图片描述
我的是cuda10.0 只能够装到1.2.0版本的pytorch, 输入安装代码（把-c pytorch删掉）

//官网的代码
//conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch
我们要输入的代码
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0