1. 输入账号密码登录
参考链接1
参考链接2
参考链接3
2. conda创建虚拟环境
conda create -n pytorch-gpu python=3.7
输入
conda env list
激活环境
conda activate 虚拟环境名
换源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
参考链接4
3. 安装torch,cuda版本
进入pytorch官网
输入
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
等待安装
python模式查看torch和cuda版本
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
之后根据需要安装自己的库,运行程序即可
4. linux使用技巧
删除 rm
编辑文件 vim 文件名称全称
保存文本 :w
退出编辑 :q
保存并退出 :wq
强制退出(不保存退出) :q!
查看隐藏文件:ls -a
复制文件到另一个目录:cp -r 文件地址1 文件地址2