今天重新装机,安装pytorch时发现一种特别简单的办法,以后再也不用为cuda纠结了
安装anaconda,
创建conda虚拟环境
激活虚拟环境
使用pip命令安装
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
使用这一条命令就可以完美安装
其中cu117为cuda版本,根据pytorch和显卡驱动而定,我的显卡驱动为515,对应cuda11.7