这里不着重讲解如何安装anconda,下面这个链接供大家参考:
(24条消息) 科研第一步:实现Linux服务器上创建python虚拟环境_fs1341825137的博客-CSDN博客
接下来anconda环境已经创建成功。接下来就是创建python环境并安装pytorch
步骤:conda create -n 虚拟环境名 python=版本号以python=3.7为例创建名字为deblur的虚拟环境
conda create -n deblur python=3.7
python环境创建之后,进行pytorch的安装。
安装pytorch之前,需要查看服务器中的cuda的版本号,因为pytorch和cuda的版本需要对应:
pytorch-cuda版本的安装:
pytorch官网版本下载
选择对应的适配,就可以使用不同的指令
使用nvcc -V查看cuda的版本号。
nvcc -V
必须使用nvcc -V查看的cuda版本号才是正确的。
然后到pytorch官网中查看相应的版本进行安装。PyTorchhttps://pytorch.org/
可以点击图片中圈出的链接,查看以前的版本。
以pytorch为1.7.1,并且cuda为10.1版本安装为例:
执行命令:
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
出现如图所示的内容说明pytorch安装成功:
进入python环境查看pytorch cude是否可以使用:
python import torch torch.cuda.is_available()
出现true说明可以使用。
注:以上步骤,需要在联网的条件下,进行安装。否则可能会不成功。



