anaconda下安装pytorch环境

1、pytorch 安装

conda create -n pytorch python=3.8

conda activate pytorch

安装 pytorch V-10.1版本

# CUDA 10.2
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

2、更改pytorch版本

如果需要更改pytorch版本，请先删除env下的pytorch环境

conda env list 查询当前env 列表

conda remove -n 环境名称 --all

3、重装anaconda

先删除envs 以及pkgs ，然后使用uninstall-anaconda.exe 进行卸载。

然后安装新的anaconda安装包。勾选添加环境变量，以及设置anaconda默认的python版本。

4、在env环境的pytorch下安装jupyter

先激活pytorch环境再进行安装。

conda activate pytorch

conda install nb_conda;

5、说明

本人显卡为1050ti，显卡驱动支持cuda：12.1。