1.查看CUDA版本
windows:控制面板->NVIDIA控制面板->系统信息
linux:cat /usr/local/cuda/version.txt
2.在官网获取pip命令
https://pytorch.org/get-started/locally/
使用镜像源加速下载 -i https://pypi.tuna.tsinghua.edu.cn/simple/
3.若linux权限不够或依赖冲突(不能用python -m venv 创建虚拟环境)——用anaconda绕过
curl -O -C- 下载地址
chmod 777 anaconda.sh
./anaconda.sh
激活anaconda自带环境:source ./anaconda安装目录/bin/activate
创建一个新的虚拟环境:python -m venv py
source ./py/bin/activate
同第二部安装pytorch
附:
1.安装位置
–target=./
2.详细安装命令
https://pytorch.org/get-started/previous-versions/