先前在论坛阅读关于如何使用Anconada安装GPU版本的pytorch,最初的一步就是创建虚拟环境
conda create -n pytorch python=3.8
然后在jupyter lab中用下面带代码打印出来一直是False
torch.cuda.is_available()
后面了解想到 jupyter不是使用Anconada创建的虚拟环境,自然无法激活GPU版本的pytorch,在这里只需要给 jupyter变更环境既可。在Anconada命令给jupyter创建一个环境:
(base) C:\Users\Administrator>conda activate pytorch
(pytorch) C:\Users\Administrator>conda install ipykernel
(pytorch) C:\Users\Administrator>python -m ipykernel install --name pytorch
然后重启 jupyter 切换内核。
当然这只是解决问题的其中一个方法。