Win10,已下载安装Python以及Anaconda、PyCharm,有独显。
查看cuda版本:
进入PyTorch官网:https://pytorch.org/ ,点击GET STARTED,选择电脑配置的对应版本,查看安装PyTorch的命令
pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
打开Anaconda Powershell Prompt,粘贴命令,下载安装即可。
在命令行输入pip show torch查看是否安装成功,以及安装信息。
同时,PyCharm也可以查看已安装的包