Anaconda中虚拟环境创建、取消——安装pytorch

创建虚拟环境

下面的detection是创建的虚拟环境的名称,python的版本是3.6

conda create -n detection python=3.6

下面的是在虚拟环境中安装pytorch,这里需要注意的是要看一下电脑里CUDA的版本,防止安装失败。

查看CUDA版本:
在命令行中输入

nvcc --version

以下同样在命令行cmd中输入,安装pytorch

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev

激活虚拟环境

这里detection是虚拟环境的名称

windows中是

conda activate detection

在Linux中则是

source activate detection

删除环境

注意删除了环境后就没法复原了

conda remove -n detection --all

退出环境

windows中是

conda deactivate

而在Linux中是

source deactivate

猜你喜欢

转载自blog.csdn.net/weixin_41729258/article/details/106185805
今日推荐