При использовании Pytorch-GPU в Anaconda, не обязательно ли устанавливать CUDA? - PullRequest
Новая
       14

При использовании Pytorch-GPU в Anaconda, не обязательно ли устанавливать CUDA?

0 голосов
/ 14 ноября 2018

Я обнаружил, что после установки версии графического процессора Pytorch 0.4 в Anaconda вам не нужно устанавливать CUDA локально, чтобы вызвать ускорение gpu.При выполнении кода ядро ​​графического процессора может использоваться более чем на 90%.

Редактировать: я использовал его в Windows 10. Не знаю, работает ли оно в Linux.

1 Ответ

0 голосов
/ 16 ноября 2018

@ talonmies

Спасибо за ваш URL. Кажется, что pytorch не нуждается в cuda в Windows, так как его зависимостями являются cffi, mkl, numpy и python.

Я ввел эту команду conda search -c pytorch pytorch=0.4.0 --info в Anaconda Prompt и там написано 

Loading channels: done
pytorch 0.4.0 py35_cuda80_cudnn7he774522_1
------------------------------------------
file name   : pytorch-0.4.0-py35_cuda80_cudnn7he774522_1.tar.bz2
name        : pytorch
version     : 0.4.0
build string: py35_cuda80_cudnn7he774522_1
build number: 1
size        : 528.5 MB
arch        : x86_64
constrains  : ()
platform    : Platform.win
license     : BSD 3-Clause
subdir      : win-64
url         : https://conda.anaconda.org/pytorch/win-64/pytorch-0.4.0-py35_cuda80_cudnn7he774522_1.tar.bz2
md5         : 7db3971bb054079d7c7ff84b6286c58e
dependencies:
  - cffi
  - mkl >=2018
  - numpy >=1.11
  - python >=3.5,<3.6.0a0


pytorch 0.4.0 py35_cuda90_cudnn7he774522_1
------------------------------------------
file name   : pytorch-0.4.0-py35_cuda90_cudnn7he774522_1.tar.bz2
name        : pytorch
version     : 0.4.0
build string: py35_cuda90_cudnn7he774522_1
build number: 1
size        : 578.5 MB
arch        : x86_64
constrains  : ()
platform    : Platform.win
license     : BSD 3-Clause
subdir      : win-64
url         : https://conda.anaconda.org/pytorch/win-64/pytorch-0.4.0-py35_cuda90_cudnn7he774522_1.tar.bz2
md5         : 8200c9841f9cad6f2e605015812aa3f2
dependencies:
  - cffi
  - mkl >=2018
  - numpy >=1.11
  - python >=3.5,<3.6.0a0


pytorch 0.4.0 py35_cuda91_cudnn7he774522_1
------------------------------------------
file name   : pytorch-0.4.0-py35_cuda91_cudnn7he774522_1.tar.bz2
name        : pytorch
version     : 0.4.0
build string: py35_cuda91_cudnn7he774522_1
build number: 1
size        : 546.1 MB
arch        : x86_64
constrains  : ()
platform    : Platform.win
license     : BSD 3-Clause
subdir      : win-64
url         : https://conda.anaconda.org/pytorch/win-64/pytorch-0.4.0-py35_cuda91_cudnn7he774522_1.tar.bz2
md5         : 79d99a825f66b55b1aa6f04d22d68aac
dependencies:
  - cffi
  - mkl >=2018
  - numpy >=1.11
  - python >=3.5,<3.6.0a0


pytorch 0.4.0 py36_cuda80_cudnn7he774522_1
------------------------------------------
file name   : pytorch-0.4.0-py36_cuda80_cudnn7he774522_1.tar.bz2
name        : pytorch
version     : 0.4.0
build string: py36_cuda80_cudnn7he774522_1
build number: 1
size        : 529.2 MB
arch        : x86_64
constrains  : ()
platform    : Platform.win
license     : BSD 3-Clause
subdir      : win-64
url         : https://conda.anaconda.org/pytorch/win-64/pytorch-0.4.0-py36_cuda80_cudnn7he774522_1.tar.bz2
md5         : 27d20c9869fb57ffe0d6d014cf348855
dependencies:
  - cffi
  - mkl >=2018
  - numpy >=1.11
  - python >=3.6,<3.7.0a0


pytorch 0.4.0 py36_cuda90_cudnn7he774522_1
------------------------------------------
file name   : pytorch-0.4.0-py36_cuda90_cudnn7he774522_1.tar.bz2
name        : pytorch
version     : 0.4.0
build string: py36_cuda90_cudnn7he774522_1
build number: 1
size        : 577.6 MB
arch        : x86_64
constrains  : ()
platform    : Platform.win
license     : BSD 3-Clause
subdir      : win-64
url         : https://conda.anaconda.org/pytorch/win-64/pytorch-0.4.0-py36_cuda90_cudnn7he774522_1.tar.bz2
md5         : 138dcca8eeff1d58a8fd9b1febf702f6
dependencies:
  - cffi
  - mkl >=2018
  - numpy >=1.11
  - python >=3.6,<3.7.0a0


pytorch 0.4.0 py36_cuda91_cudnn7he774522_1
------------------------------------------
file name   : pytorch-0.4.0-py36_cuda91_cudnn7he774522_1.tar.bz2
name        : pytorch
version     : 0.4.0
build string: py36_cuda91_cudnn7he774522_1
build number: 1
size        : 546.4 MB
arch        : x86_64
constrains  : ()
platform    : Platform.win
license     : BSD 3-Clause
subdir      : win-64
url         : https://conda.anaconda.org/pytorch/win-64/pytorch-0.4.0-py36_cuda91_cudnn7he774522_1.tar.bz2
md5         : 326265665000de6f7501160b10b089c8
dependencies:
  - cffi
  - mkl >=2018
  - numpy >=1.11
  - python >=3.6,<3.7.0a0
...