conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/lab/dev
记得换源,然后9.2这个参数要根据自己电脑配置调整,选择自己合适的版本,可以在pytorch官网找
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/lab/dev
记得换源,然后9.2这个参数要根据自己电脑配置调整,选择自己合适的版本,可以在pytorch官网找