首先查看自己的cuda和cudnn版本

两种方式:

使用pytorch查看

python
import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

直接终端查看

nvcc -V
在这里插入图片描述

安装anaconda

在Windows下的文本文件用ssh上传到Unix/Linux,会自动进行格式转换,就是把回车、换行转为换行。
所以要将文件改为zip文件后再传输,传输成功后改回sh
将.sh文件压缩打包zip传至服务器,在服务器执行

unzip Anaconda3-5.3.1-Linux-x86_64.zip
sh Anaconda3-5.3.1-Linux-x86_64.sh

一直回车
安装完成

使用anaconda

如果安装时没有选配置路径
打开.bashrc文件,在最后一行加入:

export PATH=/home/scliu/anaconda3/bin:$PATH

保存,然后打开终端运行:

source .bashrc

这样下次打开终端后就可以直接使用conda命令

安装pytorch

进入anaconda

创建虚拟环境:
conda create -n pytorch_gpu python=3.7
#查看所有虚拟环境:
conda env list
#删除虚拟环境:
conda remove -n pytorch_gpu --all
进入虚拟环境:
source activate pytorch_gpu 

去pytorch官网:
https://pytorch.org/
查看之前版本:
https://pytorch.org/get-started/previous-versions/
根据cuda版本去安装pytorch:

 conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

安装结束后查看是否安装成功:

 python
 import torch
 torch.cuda.is_available()

CommandNotFoundError: Your shell has not been properly configured to use ‘conda activate‘:
不要使用conda ativate 命令,改成source activate 命令

激活 anaconda 环境: 
source activate
退出 anaconda 环境:
source deactivate

参考anaconda环境配置:https://blog.csdn.net/weixin_40585997/article/details/114867483

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐