如何在Ubuntu OS中安装pytorch
- 创建虚拟conda环境
conda create -n conda_env_name
conda activate conda_env_name - 创建condarc文件命令
conda config --add channels r - 修改conda的国内镜像站配置文件
清华镜像源link - install pytorch, torchvision, and cuda
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
Notes:
- pytorch install document
- RTX3080仅支持cudatoolkit>10.0