实验室服务器linux环境下在虚拟环境上安装pytorch(CUDA8.0)

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

在尝试了各种方法后,终于成功在实验室服务器上安装成功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之后虚拟环境就创建成功了

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值