pytorch conda虚拟环境创建

1. 创建虚拟环境

conda create -n env_name python=3.7 #env_name:你的环境名称，自己命名

2. 查看cuda版本以便安装相配的torch版本

方法一：输入一下命令查看cuda版本

nvcc -V

方法二：

cat /usr/local/cuda/version.txt

3. pytorch下载

首先打开pytorch官网

选择上述描述与你配置相适应的条件选项，后再终端运行红框中语句进行安装。
如果上述不包含你的cuda版本或你想要别的版本的pytorch可以选择下图中位置选项进行选择

语句示例

conda install pytorch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1 cudatoolkit=10.2 -c pytorch

3. 相关包下载

需要什么就下载什么

4. 退出虚拟环境

conda deactivate

5. pytorch卸载

pip 卸载

pip uninstall torch

conda 卸载：

conda uninstall pytorch
conda uninstall libtorch

参考链接：https://blog.youkuaiyun.com/uid1547/article/details/121628288