本文主要是介绍实验室服务器linux环境下在虚拟环境上安装pytorch(CUDA8.0),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
在尝试了各种方法后,终于成功在实验室服务器上安装成功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版本(同样可以换成其他版本)&#
这篇关于实验室服务器linux环境下在虚拟环境上安装pytorch(CUDA8.0)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!