虚拟环境的配置/pytorch的安装/在Jupyter notebook 添加虚拟环境
1:创建虚拟环境,避免数据包和软件版本发生冲突:
(1)创建名为pytorch_pip 的虚拟环境
conda create -n pytorch_pip python=3.6
(2)激活虚拟环境
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版本。
(2)去pytorch官网查看与CUDA版本一致的安装命令行(在虚拟环境下安装pytorch)
conda install pytorch1.7.0 torchvision0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch
到这里的话,pytorch就在虚拟环境下安装好了
3 让Jupyter notebook 在虚拟环境下运行
(1)在要操作的虚拟环境下配置插件ipykernel
conda install ipykernel
(2) 使用ipykernel将虚拟环境添加到Jyputer notebook中
python -m ipykernel install --name Name (Name是此虚拟环境名称,可自定义)
4 在虚拟环境下启动jupyter notebook,
nohup jupyter notebook &
打开了jupyter notebook,这次就可以看到我们配置的虚拟环境了,如下图.



