PyTorch环境安装

266 阅读1分钟

1、系统:ubuntu

2、下载anaconda linux版:www.anaconda.com/products/in…

3、安装:

4、conda常用命令:

(1) conda list 查看安装了哪些包

(2)conda env list 或者 conda info -e 查看当前存在哪些虚拟环境

(3)conda update conda 检查更新当前conda

(4)conda create -n your_env_name python=X.X (2.7或者3.6) X.X为创建的python版本、名字为 your_env_name,如:conda create -n pytorch_1.7 python=3.6.

(5)conda activate your_env_name, 进入虚拟环境 conda deactivate, 退出当前虚拟环境

(6)conda install [package] 在当前虚拟环境安装某个包 conda remove [package] 在当前虚拟环境中卸载某个包

(7)conda remove -n your_env_name --all, 删除虚拟环境

5、安装PyTorch

pytorch.org/get-started…

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html