Linux下安装pytorch

本文前提:需要先安装好anaconda3,并新建环境,可以将pytorch安装在新建的环境下。

前期安装pytorch踩了很多坑,在pytorch官网下载非常慢,添加清华源镜像安装也不成功,试了很多方法,最后终于成功了!!!呜呜呜~

安装教程参考:【学习记录】Linux服务器安装Anaconda和Pytorch_哔哩哔哩_bilibili   UP主的方法一下就安装成功了!

一进入环境:

 source activate pytorch(在新建的pytorch环境中安装pytorch)

二查看cuda版本:

nvcc -V  或者 nvidia-smi      (有时两者显示cuda版本不同,原因这篇博客有解释:CUDA版本不同:nvidia-smi和nvcc -V_nvidia-smi和nvcc-v的cuda版本不一致-CSDN博客

三下载torch 阿里云镜像(快):

pip install torch torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com     (这里直接安装了)

下图安装成功截图:

四查看pytorch是否安装成功

依次执行:

python (进入python)

import torch
x = torch.zeros((2, 3, 4))
print(x)

如下图:安装成功!!!

  • 7
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值