ubuntu电脑安装GPU版pytorch经验
步骤:
1、安装GPU显卡驱动
ubuntu-drivers devices
会列出本机显卡型号和recommended 驱动;然后运行
sudo ubuntu-drivers auto install
自行安装驱动;其后运行
nvidia-smi
可列出相关显卡信息。
2、下载安装miniconda3 / anaconda sh文件
下载miniconda3 sh文件,网址是 Miniconda — Conda documentationhttps://docs.conda.io/en/latest/miniconda.html
运行 命令安装:
sh miniconda3_name.sh
3、新建 虚拟环境
conda create -n torch_envname py