安装pytorch.cuda
时间: 2023-11-14 07:10:57 浏览: 123
win10+cuda10.0+pytorch安装
5星 · 资源好评率100%
安装pytorch.cuda需要以下步骤:
1. 在anaconda prompt中创建一个新的环境,例如:conda create -n pytorch_env
2. 激活该环境,例如:conda activate pytorch_env
3. 根据你的CUDA版本和pytorch版本选择相应的安装指令,例如:conda install pytorch torchvision cudatoolkit=10.1
4. 如果你想从清华源下载相应的包,需要去掉-c pytorch,例如:conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
5. 如果你想使用pip安装,可以使用以下指令:pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116 -i https://pypi.tuna.tsinghua.edu.***. 如何更新pytorch?
阅读全文