当前位置:   article > 正文

anoconda下安装pytorch_anocada torch

anocada torch

参考博客

https://blog.csdn.net/gzygwp/article/details/99719253?utm_medium=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-BlogCommendFromMachineLearnPai2-1.channel_param

CUDA 和cudnn 的安装
cuda安装
cuda 10.1
在这里插入图片描述
驱动如果比要安装的版本新,则取消版本驱动安装。
在这里插入图片描述
在这里插入图片描述
cuda 安装成功
在这里插入图片描述

Cudnn安装
下载cudnn,和cuda 版本要一致

在这里插入图片描述

在这里插入图片描述

将其更名解压到cuda 安装路径下
在这里插入图片描述

Pytorch和torchvision 的安装
2.1 Pytorch安装
这里不太推荐按照官网给的指令下载,下载很慢,而且容易出问题
2.1.1 普通方法:
首先进入 Pytorch 官网:https://pytorch.org/
点击Get Started, 选择合适的版本
在这里插入图片描述
按照给出的指令输入到anaconda promopt里输入
activate pytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
虽然是官方方法,但是不论用conda还是pip安装都会很慢,在此推荐第二种方法

2.1.2 镜像下载安装(强烈推荐)
首先进入此链接 torch清华镜像源链接.
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
pytorch-1.60 python 3.6 cuda 10.1 cndnn7.0 .bz
在这里插入图片描述
进入到安装包所在文件夹
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/盐析白兔/article/detail/150337
推荐阅读
相关标签
  

闽ICP备14008679号