search

openai / jukebox

Code for the paper "Jukebox: A Generative Model for Music"
https://openai.com/blog/jukebox/
Other
7.75k stars 1.4k forks source link

Fresh install - ModuleNotFoundError: No module named 'torch' #127

Open DMon19 opened 4 years ago

DMon19 commented 4 years ago

Hi all,

Seems to be failing due to the torch dependency, because of some conflicts. Never worked in any python environment before so installed everything fresh here. Tried the whole installation first time, failed for same reason. Deleted the entire folder to start again with the following logs.

Cheers for the help. 

(base) C:\AI\Jukebox>conda create --name jukebox python=3.7.5
WARNING: A conda environment already exists at 'C:\Users\Dave\miniconda3\envs\jukebox'
Remove existing environment (y/[n])? y

Collecting package metadata (current_repodata.json): done
Solving environment: failed with repodata from current_repodata.json, will retry with next repodata source.
Collecting package metadata (repodata.json): done
Solving environment: done

## Package Plan ##

  environment location: C:\Users\Dave\miniconda3\envs\jukebox

  added / updated specs:
    - python=3.7.5

The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    openssl-1.1.1g             |       he774522_0         4.8 MB
    ------------------------------------------------------------
                                           Total:         4.8 MB

The following NEW packages will be INSTALLED:

  ca-certificates    pkgs/main/win-64::ca-certificates-2020.6.24-0
  certifi            pkgs/main/win-64::certifi-2020.6.20-py37_0
  openssl            pkgs/main/win-64::openssl-1.1.1g-he774522_0
  pip                pkgs/main/win-64::pip-20.1.1-py37_1
  python             pkgs/main/win-64::python-3.7.5-h8c8aaf0_0
  setuptools         pkgs/main/win-64::setuptools-49.2.0-py37_0
  sqlite             pkgs/main/win-64::sqlite-3.32.3-h2a8f88b_0
  vc                 pkgs/main/win-64::vc-14.1-h0510ff6_4
  vs2015_runtime     pkgs/main/win-64::vs2015_runtime-14.16.27012-hf0eaf9b_3
  wheel              pkgs/main/win-64::wheel-0.34.2-py37_0
  wincertstore       pkgs/main/win-64::wincertstore-0.2-py37_0
  zlib               pkgs/main/win-64::zlib-1.2.11-h62dcd97_4

Proceed ([y]/n)? y

Downloading and Extracting Packages
openssl-1.1.1g       | 4.8 MB    | ############################################################################ | 100%
Preparing transaction: done
Verifying transaction: done
Executing transaction: done
#
# To activate this environment, use
#
#     $ conda activate jukebox
#
# To deactivate an active environment, use
#
#     $ conda deactivate

(base) C:\AI\Jukebox>conda activate jukebox

(jukebox) C:\AI\Jukebox>conda install mpi4py=3.0.3
Collecting package metadata (current_repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Collecting package metadata (repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.

PackagesNotFoundError: The following packages are not available from current channels:

  - mpi4py=3.0.3

Current channels:

  - https://repo.anaconda.com/pkgs/main/win-64
  - https://repo.anaconda.com/pkgs/main/noarch
  - https://repo.anaconda.com/pkgs/r/win-64
  - https://repo.anaconda.com/pkgs/r/noarch
  - https://repo.anaconda.com/pkgs/msys2/win-64
  - https://repo.anaconda.com/pkgs/msys2/noarch

To search for alternate channels that may provide the conda package you're
looking for, navigate to

    https://anaconda.org

and use the search bar at the top of the page.

(jukebox) C:\AI\Jukebox>pip install mpi4py==3.0.3
Collecting mpi4py==3.0.3
  Using cached mpi4py-3.0.3-cp37-cp37m-win_amd64.whl (477 kB)
Installing collected packages: mpi4py
Successfully installed mpi4py-3.0.3

(jukebox) C:\AI\Jukebox>conda install pytorch=1.4 torchvision=0.5 cudatoolkit=10.0 -c pytorch
Collecting package metadata (current_repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Solving environment: failed with repodata from current_repodata.json, will retry with next repodata source.
Collecting package metadata (repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Solving environment: /
Found conflicts! Looking for incompatible packages.
This can take several minutes.  Press CTRL-C to abort.
failed

UnsatisfiableError: The following specifications were found to be incompatible with each other:

Output in format: Requested package -> Available versions

Package zlib conflicts for:
python=3.7.5 -> sqlite[version='>=3.30.1,<4.0a0'] -> zlib[version='>=1.2.11,<1.3.0a0']
torchvision=0.5 -> pillow[version='>=4.1.1'] -> zlib[version='>=1.2.11,<1.3.0a0']
zlib
sqlite -> zlib[version='>=1.2.11,<1.3.0a0']

Package openssl conflicts for:
python=3.7.5 -> openssl[version='>=1.1.1d,<1.1.2a']
openssl
pytorch=1.4 -> python[version='>=3.8,<3.9.0a0'] -> openssl[version='>=1.1.1a,<1.1.2a|>=1.1.1b,<1.1.2a|>=1.1.1c,<1.1.2a|>=1.1.1d,<1.1.2a|>=1.1.1e,<1.1.2a|>=1.1.1f,<1.1.2a|>=1.1.1g,<1.1.2a']
wincertstore -> python[version='>=3.8,<3.9.0a0'] -> openssl[version='>=1.1.1a,<1.1.2a|>=1.1.1b,<1.1.2a|>=1.1.1c,<1.1.2a|>=1.1.1d,<1.1.2a|>=1.1.1e,<1.1.2a|>=1.1.1f,<1.1.2a|>=1.1.1g,<1.1.2a']
wheel -> python[version='>=3.8,<3.9.0a0'] -> openssl[version='>=1.1.1a,<1.1.2a|>=1.1.1b,<1.1.2a|>=1.1.1c,<1.1.2a|>=1.1.1d,<1.1.2a|>=1.1.1e,<1.1.2a|>=1.1.1f,<1.1.2a|>=1.1.1g,<1.1.2a']
setuptools -> python[version='>=3.8,<3.9.0a0'] -> openssl[version='>=1.1.1a,<1.1.2a|>=1.1.1b,<1.1.2a|>=1.1.1c,<1.1.2a|>=1.1.1d,<1.1.2a|>=1.1.1e,<1.1.2a|>=1.1.1f,<1.1.2a|>=1.1.1g,<1.1.2a']
certifi -> python[version='>=3.7,<3.8.0a0'] -> openssl[version='>=1.1.1a,<1.1.2a|>=1.1.1b,<1.1.2a|>=1.1.1c,<1.1.2a|>=1.1.1d,<1.1.2a|>=1.1.1e,<1.1.2a|>=1.1.1f,<1.1.2a|>=1.1.1g,<1.1.2a']
torchvision=0.5 -> python[version='>=3.8,<3.9.0a0'] -> openssl[version='>=1.1.1a,<1.1.2a|>=1.1.1b,<1.1.2a|>=1.1.1c,<1.1.2a|>=1.1.1d,<1.1.2a|>=1.1.1e,<1.1.2a|>=1.1.1f,<1.1.2a|>=1.1.1g,<1.1.2a']
pip -> python[version='>=3.7,<3.8.0a0'] -> openssl[version='>=1.1.1a,<1.1.2a|>=1.1.1b,<1.1.2a|>=1.1.1c,<1.1.2a|>=1.1.1d,<1.1.2a|>=1.1.1e,<1.1.2a|>=1.1.1f,<1.1.2a|>=1.1.1g,<1.1.2a']

Package six conflicts for:
pytorch=1.4 -> mkl-service[version='>=2,<3.0a0'] -> six
pip -> html5lib -> six[version='>=1.9']
torchvision=0.5 -> six

Package vs2008_runtime conflicts for:
zlib -> vc=9 -> vs2008_runtime[version='>=9.0.30729.1,<10.0a0']
certifi -> python[version='>=2.7,<2.8.0a0'] -> vs2008_runtime
openssl -> vc=9 -> vs2008_runtime[version='>=9.0.30729.1,<10.0a0']
wheel -> python[version='>=2.7,<2.8.0a0'] -> vs2008_runtime
sqlite -> vc=9 -> vs2008_runtime[version='>=9.0.30729.1,<10.0a0']
wincertstore -> python[version='>=2.7,<2.8.0a0'] -> vs2008_runtime
setuptools -> python[version='>=2.7,<2.8.0a0'] -> vs2008_runtime
vc -> vs2008_runtime[version='>=9.0.30729.1,<10.0a0']
pip -> python[version='>=2.7,<2.8.0a0'] -> vs2008_runtime

Package wincertstore conflicts for:
setuptools -> wincertstore[version='>=0.2']
wincertstore
wheel -> setuptools -> wincertstore[version='>=0.2']
pip -> setuptools -> wincertstore[version='>=0.2']

Package setuptools conflicts for:
pip -> setuptools
setuptools
wheel -> setuptools
python=3.7.5 -> pip -> setuptools

Package vs2015_runtime conflicts for:
pytorch=1.4 -> python[version='>=3.6,<3.7.0a0'] -> vs2015_runtime[version='>=14.15.26706|>=14.16.27012']
sqlite -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
python=3.7.5 -> vc[version='>=14.1,<15.0a0'] -> vs2015_runtime[version='>=14.15.26706|>=14.16.27012']
python=3.7.5 -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
openssl -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
pip -> python[version='>=3.7,<3.8.0a0'] -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
torchvision=0.5 -> numpy[version='>=1.11'] -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
vc -> vs2015_runtime[version='>=14.0.25123,<15.0a0|>=14.0.25420|>=14.15.26706|>=14.16.27012']
vs2015_runtime
zlib -> vc[version='>=14.1,<15.0a0'] -> vs2015_runtime[version='>=14.0.25123,<15.0a0|>=14.0.25420|>=14.15.26706|>=14.16.27012']
zlib -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
wheel -> python[version='>=3.6,<3.7.0a0'] -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
certifi -> python[version='>=3.7,<3.8.0a0'] -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
sqlite -> vc[version='>=14.1,<15.0a0'] -> vs2015_runtime[version='>=14.0.25123,<15.0a0|>=14.0.25420|>=14.15.26706|>=14.16.27012']
pytorch=1.4 -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
openssl -> vc[version='>=14.1,<15.0a0'] -> vs2015_runtime[version='>=14.0.25123,<15.0a0|>=14.0.25420|>=14.15.26706|>=14.16.27012']
wincertstore -> python[version='>=3.8,<3.9.0a0'] -> vs2015_runtime[version='>=14.16.27012,<15.0a0']
setuptools -> python[version='>=3.8,<3.9.0a0'] -> vs2015_runtime[version='>=14.16.27012,<15.0a0']

Package wheel conflicts for:
wheel
pip -> wheel
python=3.7.5 -> pip -> wheel

Package certifi conflicts for:
pip -> setuptools -> certifi[version='>=2016.09|>=2016.9.26|>=2017.4.17']
wheel -> setuptools -> certifi[version='>=2016.09|>=2016.9.26']
setuptools -> certifi[version='>=2016.09|>=2016.9.26']
certifi

Package sqlite conflicts for:
python=3.7.5 -> sqlite[version='>=3.30.1,<4.0a0']
torchvision=0.5 -> python[version='>=3.8,<3.9.0a0'] -> sqlite[version='>=3.25.3,<4.0a0|>=3.26.0,<4.0a0|>=3.27.2,<4.0a0|>=3.28.0,<4.0a0|>=3.29.0,<4.0a0|>=3.30.0,<4.0a0|>=3.30.1,<4.0a0|>=3.31.1,<4.0a0|>=3.32.3,<4.0a0']
wheel -> python[version='>=3.6,<3.7.0a0'] -> sqlite[version='>=3.25.3,<4.0a0|>=3.26.0,<4.0a0|>=3.29.0,<4.0a0|>=3.30.1,<4.0a0|>=3.31.1,<4.0a0|>=3.32.3,<4.0a0|>=3.30.0,<4.0a0|>=3.28.0,<4.0a0|>=3.27.2,<4.0a0']
certifi -> python[version='>=3.7,<3.8.0a0'] -> sqlite[version='>=3.25.3,<4.0a0|>=3.26.0,<4.0a0|>=3.27.2,<4.0a0|>=3.28.0,<4.0a0|>=3.29.0,<4.0a0|>=3.30.1,<4.0a0|>=3.31.1,<4.0a0|>=3.32.3,<4.0a0|>=3.30.0,<4.0a0']
setuptools -> python[version='>=3.8,<3.9.0a0'] -> sqlite[version='>=3.25.3,<4.0a0|>=3.26.0,<4.0a0|>=3.27.2,<4.0a0|>=3.28.0,<4.0a0|>=3.29.0,<4.0a0|>=3.30.0,<4.0a0|>=3.30.1,<4.0a0|>=3.31.1,<4.0a0|>=3.32.3,<4.0a0']
wincertstore -> python[version='>=3.8,<3.9.0a0'] -> sqlite[version='>=3.25.3,<4.0a0|>=3.26.0,<4.0a0|>=3.27.2,<4.0a0|>=3.28.0,<4.0a0|>=3.29.0,<4.0a0|>=3.30.0,<4.0a0|>=3.30.1,<4.0a0|>=3.31.1,<4.0a0|>=3.32.3,<4.0a0']
sqlite
pip -> python[version='>=3.7,<3.8.0a0'] -> sqlite[version='>=3.25.3,<4.0a0|>=3.26.0,<4.0a0|>=3.27.2,<4.0a0|>=3.28.0,<4.0a0|>=3.29.0,<4.0a0|>=3.30.1,<4.0a0|>=3.31.1,<4.0a0|>=3.32.3,<4.0a0|>=3.30.0,<4.0a0']
pytorch=1.4 -> python[version='>=3.8,<3.9.0a0'] -> sqlite[version='>=3.25.3,<4.0a0|>=3.26.0,<4.0a0|>=3.27.2,<4.0a0|>=3.28.0,<4.0a0|>=3.29.0,<4.0a0|>=3.30.0,<4.0a0|>=3.30.1,<4.0a0|>=3.31.1,<4.0a0|>=3.32.3,<4.0a0']

Package vc conflicts for:
certifi -> python[version='>=3.7,<3.8.0a0'] -> vc[version='14.*|>=14.1,<15.0a0|9.*']
openssl -> vc[version='14.*|9.*|>=14.1,<15.0a0']
setuptools -> python[version='>=3.8,<3.9.0a0'] -> vc[version='14.*|>=14.1,<15.0a0|9.*']
python=3.7.5 -> vc[version='>=14.1,<15.0a0']
wheel -> python[version='>=3.6,<3.7.0a0'] -> vc[version='14.*|>=14.1,<15.0a0|9.*']
pytorch=1.4 -> vc[version='>=14.1,<15.0a0']
sqlite -> vc[version='14.*|9.*|>=14.1,<15.0a0']
vc
pip -> python[version='>=3.7,<3.8.0a0'] -> vc[version='14.*|>=14.1,<15.0a0|9.*']
python=3.7.5 -> openssl[version='>=1.1.1d,<1.1.2a'] -> vc=9
pytorch=1.4 -> ninja -> vc[version='14.*|9.*']
wincertstore -> python[version='>=3.8,<3.9.0a0'] -> vc[version='14.*|>=14.1,<15.0a0|9.*']
zlib -> vc[version='14.*|9.*|>=14.1,<15.0a0']
torchvision=0.5 -> numpy[version='>=1.11'] -> vc[version='14.*|9.*|>=14.1,<15.0a0']

Package ca-certificates conflicts for:
pip -> python[version='>=2.7,<2.8.0a0'] -> ca-certificates
setuptools -> python[version='>=2.7,<2.8.0a0'] -> ca-certificates
certifi -> python[version='>=2.7,<2.8.0a0'] -> ca-certificates
openssl -> ca-certificates
ca-certificates
wheel -> python[version='>=2.7,<2.8.0a0'] -> ca-certificates
wincertstore -> python[version='>=2.7,<2.8.0a0'] -> ca-certificates
python=3.7.5 -> openssl[version='>=1.1.1d,<1.1.2a'] -> ca-certificates

Package cudatoolkit conflicts for:
pytorch=1.4 -> cudatoolkit[version='>=10.1,<10.2|>=9.2,<9.3']
cudatoolkit=10.0
torchvision=0.5 -> cudatoolkit[version='>=10.1,<10.2|>=9.2,<9.3']

Package pytorch conflicts for:
pytorch=1.4
torchvision=0.5 -> pytorch==1.4.0

Package pip conflicts for:
wheel -> python[version='>=3.6,<3.7.0a0'] -> pip
pip
wincertstore -> python[version='>=3.8,<3.9.0a0'] -> pip
pytorch=1.4 -> python[version='>=3.8,<3.9.0a0'] -> pip
setuptools -> python[version='>=3.8,<3.9.0a0'] -> pip
python=3.7.5 -> pip
certifi -> python[version='>=3.7,<3.8.0a0'] -> pip
torchvision=0.5 -> python[version='>=3.8,<3.9.0a0'] -> pip

Package webencodings conflicts for:
python=3.7.5 -> pip -> webencodings
pip -> webencodingsThe following specifications were found to be incompatible with your CUDA driver:

  - feature:/win-64::__cuda==11.0=0
  - feature:|@/win-64::__cuda==11.0=0

Your installed CUDA driver is: 11.0

(jukebox) C:\AI\Jukebox>git clone https://github.com/openai/jukebox.git
Cloning into 'jukebox'...
remote: Enumerating objects: 910, done.
Receiving objects: 100% (910/910), 2.74 MiB | 3.75 MiB/s, done.
remote: Total 910 (delta 0), reused 0 (delta 0), pack-reused 910
Resolving deltas: 100% (444/444), done.

(jukebox) C:\AI\Jukebox>cd jukebox

(jukebox) C:\AI\Jukebox\jukebox>pip install -r requirements.txt
Processing c:\users\dave\appdata\local\pip\cache\wheels\dd\c5\df\d9bf8223023d31343b65f1cc57d2dc005610ebbcd2b4a5d1e7\fire-0.1.3-py2.py3-none-any.whl
Collecting tqdm==4.45.0
  Using cached tqdm-4.45.0-py2.py3-none-any.whl (60 kB)
Collecting soundfile==0.10.3.post1
  Using cached SoundFile-0.10.3.post1-py2.py3.cp26.cp27.cp32.cp33.cp34.cp35.cp36.pp27.pp32.pp33-none-win_amd64.whl (689 kB)
Collecting unidecode==1.1.1
  Using cached Unidecode-1.1.1-py2.py3-none-any.whl (238 kB)
Collecting numba==0.48.0
  Using cached numba-0.48.0-cp37-cp37m-win_amd64.whl (2.1 MB)
Processing c:\users\dave\appdata\local\pip\cache\wheels\18\9e\42\3224f85730f92fa2925f0b4fb6ef7f9c5431a64dfc77b95b39\librosa-0.7.2-py3-none-any.whl
Requirement already satisfied: mpi4py>=3.0.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from -r requirements.txt (line 7)) (3.0.3)
Collecting six
  Using cached six-1.15.0-py2.py3-none-any.whl (10 kB)
Collecting cffi>=1.0
  Using cached cffi-1.14.0-cp37-cp37m-win_amd64.whl (176 kB)
Requirement already satisfied: setuptools in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from numba==0.48.0->-r requirements.txt (line 5)) (49.2.0.post20200714)
Collecting numpy>=1.15
  Using cached numpy-1.19.0-cp37-cp37m-win_amd64.whl (13.0 MB)
Collecting llvmlite<0.32.0,>=0.31.0dev0
  Using cached llvmlite-0.31.0-cp37-cp37m-win_amd64.whl (13.6 MB)
Processing c:\users\dave\appdata\local\pip\cache\wheels\a0\18\0a\8ad18a597d8333a142c9789338a96a6208f1198d290ece356c\resampy-0.2.2-py3-none-any.whl
Collecting decorator>=3.0.0
  Using cached decorator-4.4.2-py2.py3-none-any.whl (9.2 kB)
Processing c:\users\dave\appdata\local\pip\cache\wheels\cd\85\04\704a589ef8773d1ec1393c4b271a023f929d85ef5317b12d8c\audioread-2.1.8-py3-none-any.whl
Collecting scikit-learn!=0.19.0,>=0.14.0
  Using cached scikit_learn-0.23.1-cp37-cp37m-win_amd64.whl (6.8 MB)
Collecting scipy>=1.0.0
  Using cached scipy-1.5.1-cp37-cp37m-win_amd64.whl (31.2 MB)
Collecting joblib>=0.12
  Using cached joblib-0.16.0-py3-none-any.whl (300 kB)
Collecting pycparser
  Using cached pycparser-2.20-py2.py3-none-any.whl (112 kB)
Collecting threadpoolctl>=2.0.0
  Using cached threadpoolctl-2.1.0-py3-none-any.whl (12 kB)
Installing collected packages: six, fire, tqdm, pycparser, cffi, soundfile, unidecode, numpy, llvmlite, numba, scipy, resampy, decorator, audioread, threadpoolctl, joblib, scikit-learn, librosa
Successfully installed audioread-2.1.8 cffi-1.14.0 decorator-4.4.2 fire-0.1.3 joblib-0.16.0 librosa-0.7.2 llvmlite-0.31.0 numba-0.48.0 numpy-1.19.0 pycparser-2.20 resampy-0.2.2 scikit-learn-0.23.1 scipy-1.5.1 six-1.15.0 soundfile-0.10.3.post1 threadpoolctl-2.1.0 tqdm-4.45.0 unidecode-1.1.1

(jukebox) C:\AI\Jukebox\jukebox>pip install -e .
Obtaining file:///C:/AI/Jukebox/jukebox
Requirement already satisfied: fire==0.1.3 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from jukebox==1.0) (0.1.3)
Requirement already satisfied: tqdm==4.45.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from jukebox==1.0) (4.45.0)
Requirement already satisfied: soundfile==0.10.3.post1 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from jukebox==1.0) (0.10.3.post1)
Requirement already satisfied: unidecode==1.1.1 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from jukebox==1.0) (1.1.1)
Requirement already satisfied: numba==0.48.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from jukebox==1.0) (0.48.0)
Requirement already satisfied: librosa==0.7.2 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from jukebox==1.0) (0.7.2)
Requirement already satisfied: mpi4py>=3.0.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from jukebox==1.0) (3.0.3)
Requirement already satisfied: six in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from fire==0.1.3->jukebox==1.0) (1.15.0)
Requirement already satisfied: cffi>=1.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from soundfile==0.10.3.post1->jukebox==1.0) (1.14.0)
Requirement already satisfied: llvmlite<0.32.0,>=0.31.0dev0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from numba==0.48.0->jukebox==1.0) (0.31.0)
Requirement already satisfied: setuptools in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from numba==0.48.0->jukebox==1.0) (49.2.0.post20200714)
Requirement already satisfied: numpy>=1.15 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from numba==0.48.0->jukebox==1.0) (1.19.0)
Requirement already satisfied: joblib>=0.12 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from librosa==0.7.2->jukebox==1.0) (0.16.0)
Requirement already satisfied: decorator>=3.0.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from librosa==0.7.2->jukebox==1.0) (4.4.2)
Requirement already satisfied: audioread>=2.0.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from librosa==0.7.2->jukebox==1.0) (2.1.8)
Requirement already satisfied: scikit-learn!=0.19.0,>=0.14.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from librosa==0.7.2->jukebox==1.0) (0.23.1)
Requirement already satisfied: scipy>=1.0.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from librosa==0.7.2->jukebox==1.0) (1.5.1)
Requirement already satisfied: resampy>=0.2.2 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from librosa==0.7.2->jukebox==1.0) (0.2.2)
Requirement already satisfied: pycparser in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from cffi>=1.0->soundfile==0.10.3.post1->jukebox==1.0) (2.20)
Requirement already satisfied: threadpoolctl>=2.0.0 in c:\users\dave\miniconda3\envs\jukebox\lib\site-packages (from scikit-learn!=0.19.0,>=0.14.0->librosa==0.7.2->jukebox==1.0) (2.1.0)
Installing collected packages: jukebox
  Running setup.py develop for jukebox
Successfully installed jukebox

(jukebox) C:\AI\Jukebox\jukebox>python jukebox/sample.py --model=5b_lyrics --name=sample_5b_prompted --levels=3 --mode=primed \
Traceback (most recent call last):
  File "jukebox/sample.py", line 2, in <module>
    import torch as t
ModuleNotFoundError: No module named 'torch'
johndpope commented 3 years ago

conda create --name jukebox python=3.7.5

it seems like the python you've chosen here is unusual https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-python.html

if you use python=3.7 does it help? maybe your pip / python / pip3 / python3 got messed up. if you type which python and python --version what happens? it seems to missing pytorch

what graphics card do you have nvidia-smi

cuda toolkit=10.2 the cudatoolkit should match up with your drivers