本次使用conda进行安装。
新建环境后,激活该环境,(记得换成清华源)使用下面的语句指定cuda版本9.0,
conda install pytorch torchvision cudatoolkit=9.0
安装后,一句代码能够安装全部需要的版本对应依赖。你只需要根据自己的显卡驱动指定cuda的版本即可。
本次使用conda进行安装。
新建环境后,激活该环境,(记得换成清华源)使用下面的语句指定cuda版本9.0,
conda install pytorch torchvision cudatoolkit=9.0
安装后,一句代码能够安装全部需要的版本对应依赖。你只需要根据自己的显卡驱动指定cuda的版本即可。