我尝试过多种方法,例如
conda install -y pytorch==1.9 torchvision torchaudio torchtext cudatoolkit=11.0 -c pytorch -c nvidia
Run Code Online (Sandbox Code Playgroud)
但由于某种原因,它永远不会下载 cuda 11.0 或更高版本的版本。
错误消息太大而无法粘贴,但您可以在此处查看详细信息:https://github.com/pytorch/text/issues/1395
使用空环境应该很容易重现,如下所示:
conda create -n env_a40 python=3.9
conda activate env_a40
conda install -y pytorch==1.9 torchvision torchaudio torchtext cudatoolkit=11.0 -c pytorch -c nvidia
Run Code Online (Sandbox Code Playgroud)
交叉发布:
有关的:
请注意,您也可以使用 pip 尝试:
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
Run Code Online (Sandbox Code Playgroud)
还没有成功。
pytorch ×1