环境安装
conda环境安装
- 访问网址:Download Anaconda Distribution,填写邮箱,点击提交,复制下载链接。
- 在Linux上输入wget https://xxxx.sh,进行下载
- 下载过后,输入bash Anaconda3_xxx.sh
- 会有个命令行界面提供安装指南,阅读许可,按Enter阅读,然后输入yes接受条款,选择安装位置。
- 问你是否初始化,选择yes。
创建环境
- 输入conda info -e
- 输入conda create -n py310 python=3.10
- 输入conda activate py310
安装pytorch
- 安装cuda驱动,有nvidia-msi
- 驱动和cuda toolkit版本对应:CUDA 12.5 Update 1 Release Notes (nvidia.com),
CUDA Toolkit 12.5 Update 1 Downloads | NVIDIA Developer - 下面有个Archive of Previous CUDA Releases,点击可跳转查找历史版本。
- 安装cuda toolkit,有nvcc -V,
安装cudnn,(这个好像不需要单独装,安装pytorch过程中会自动安装这个) - 过程中如果需要换源:
mv /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.bak
wget -O /etc/yum.repos.d/CentOS-Base.repo https://mirrors.aliyun.com/repo/Centos-7.repo
杂项问题解决
修改DNS
vim /etc/resolv.conf
pip install安装包添加代理
pip install --proxy=http://192.168.2.168:7890 torch_xxx.whl