使用unbuntu18.04在anaconda中创建独立环境,并安装pytorch
一、创建环境
conda create -n pytorch python=3.5
(pytorch是创建的环境名称,python=3.5是设置python版本)
二、启动环境
- 输入以下命令启动环境
conda activate pytorch
输入以下命令关闭环境
conda deactivate
- 若启动环境失败,尝试用以下命令启动
激活环境
source activate pytorch
退出环境
source deactivate
三、查看电脑CUDA版本,若电脑只有CPU则忽略这一步
打开终端,输入:
nvidia-smi
可以看到右上角的CUDA Version为10.2
四、安装pytorch
- 点击以下链接,进入pytorch官网下载
2.按照上面选择,复制命令:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
启动环境后,粘贴命令运行:
输入y,等待安装完成即可。