问题:
在安装pytorch历史版本时,已经选择了GPU版本安装命令,但安装以后却总是无法在GPU上运行。
办法:
1. 选择正确的pytorch安装命令(pytorch版本正确、非CPU版本);
2. 在安装命令后添加 -c nvidia,然后正常安装。
示例:
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c nvidia
问题:
在安装pytorch历史版本时,已经选择了GPU版本安装命令,但安装以后却总是无法在GPU上运行。
办法:
1. 选择正确的pytorch安装命令(pytorch版本正确、非CPU版本);
2. 在安装命令后添加 -c nvidia,然后正常安装。
示例:
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c nvidia