在尝试了各种方法后,终于成功在实验室服务器上安装成功Pytorch,下面分享一下自己的过程和一些避雷:
1、首先安装anaconda比较简单
2、在anaconda上使用清华源,这样下载速度会大大提升,不然下载太慢一晚上都下不完,具体步骤可以参考:
https://blog.csdn.net/WannaSeaU/article/details/88427010
3、在Linux上输入以下指令查看CUDA的版本,因为安装pytorch要对应CUDA版本(如果没有CUDA则安装无CUDA版本的)
第一种:nvcc -V 或 nvcc --version(我一开始输入nvcc -V不行,换成version就可以了奇奇怪怪)
第二种:cat/usr/local/cuda/version.txt
输入之后,可以查看到当前实验室服务器的CUDA版本,如图所示,我的服务器上CUDA是8.0.61版本:

4、现在我们创建个虚拟环境来安装pytorch,避免之后要安装不同的版本等造成混乱
在命令行输入指令:conda create -n pytorch python=3.6 #将虚拟环境命名为pytorch(也可以命名其他,不影响),环境是用的是python3.6版本(同样可以换成其他版本),回车之后会开始安装,点击yes之后虚拟环境就创建成功了

本文介绍了如何在Linux服务器中,通过Anaconda在虚拟环境中成功安装对应CUDA8.0版本的PyTorch。关键步骤包括使用清华源加速下载,检查CUDA版本,创建并激活虚拟环境,以及遵循旧版本PyTorch的安装指南。最后,通过运行测试代码验证安装是否成功。
&spm=1001.2101.3001.5002&articleId=109112959&d=1&t=3&u=56811e6c63bb477896d214d453ba004e)
25万+

被折叠的 条评论
为什么被折叠?



