ubuntu pytorch install

nvidia 驱动安装

https://www.cnblogs.com/lif323/p/17014199.html

conda 安装

下载.sh

到该网站下载需要的.sh文件

wget https://repo.anaconda.com/archive/Anaconda3-2022.10-Linux-x86_64.sh

运行安装脚本

bash Anaconda3-2022.10-Linux-x86_64.sh

安装pytorch

到官网选择匹配当前环境的安装命令。

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia