当前位置:   article > 正文

Linux下安装pytorch_linux pytorch

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)

如下图:安装成功!!!

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

闽ICP备14008679号