配置虚拟环境报错
conda create -n tf37 python=3.7
conda activate tf37
conda deactivate
按照pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
在创建的虚拟环境中运行上述语句,之后进入python
说明torch安装ok了,并且可用gpu。
删除虚拟环境:
conda env remove --name your_env_name
换环境安装时候会报错:
换为
pip3 install torch torchvision torchaudio
即可。