赞
踩
本文前提:需要先安装好anaconda3,并新建环境,可以将pytorch安装在新建的环境下。
前期安装pytorch踩了很多坑,在pytorch官网下载非常慢,添加清华源镜像安装也不成功,试了很多方法,最后终于成功了!!!呜呜呜~
安装教程参考:【学习记录】Linux服务器安装Anaconda和Pytorch_哔哩哔哩_bilibili UP主的方法一下就安装成功了!
source activate pytorch(在新建的pytorch环境中安装pytorch)
nvcc -V 或者 nvidia-smi (有时两者显示cuda版本不同,原因这篇博客有解释:CUDA版本不同:nvidia-smi和nvcc -V_nvidia-smi和nvcc-v的cuda版本不一致-CSDN博客)
pip install torch torchvision -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com (这里直接安装了)
下图安装成功截图:
依次执行:
python (进入python)
import torch
x = torch.zeros((2, 3, 4))
print(x)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。