虚拟环境的配置/pytorch的安装/在Jupyter notebook 添加虚拟环境

虚拟环境的配置/pytorch的安装/在Jupyter notebook 添加虚拟环境

1:创建虚拟环境,避免数据包和软件版本发生冲突:
(1)创建虚拟环境和下载python (pytorch, python 和CUDA的版本要配对使用)
(版本配对信息见:https://blog.csdn.net/qq_29960631/article/details/121849017)
conda create -n env_name python=3.6
(2)激活虚拟环境pytorch_pip
source activate pytorch_pip
(3)退出当前虚拟环境
conda deactivate
(4) Python为什么要使用虚拟环境-Python虚拟环境的安装和配置(https://blog.csdn.net/godot06/article/details/105273998)

2:安装pytorch
(1)查看cuda的版本
nvidia-smi显示的是编译版本,nvcc -V查看的运行时cuda版本。
虚拟环境的配置/pytorch的安装/在Jupyter notebook 添加虚拟环境_第1张图片

(2)去pytorch官网查看与CUDA版本一致的安装命令行,并在虚拟环境下安装pytorch
conda install pytorch1.7.0 torchvision0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch
虚拟环境的配置/pytorch的安装/在Jupyter notebook 添加虚拟环境_第2张图片

到这里的话,pytorch就在虚拟环境下安装好了

3 让Jupyter notebook 在虚拟环境下运行
(1)在要操作的虚拟环境下配置插件ipykernel
conda install ipykernel
(2) 使用ipykernel将虚拟环境添加到Jyputer notebook中
python -m ipykernel install --name Name --user(Name是此虚拟环境名称,可自定义)

4 在虚拟环境下启动jupyter notebook,
nohup jupyter notebook &
打开了jupyter notebook,这次就可以看到我们配置的虚拟环境了,如下图.
虚拟环境的配置/pytorch的安装/在Jupyter notebook 添加虚拟环境_第3张图片

你可能感兴趣的:(pytorch)