Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add channel conda-forge to environment.yml for Windows support #80

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

SetZero
Copy link

@SetZero SetZero commented Dec 4, 2021

Currently the nvidia channel is missing cudatoolkit in version 11.1 for Windows. (https://anaconda.org/nvidia/cudatoolkit/files)
Channel conda-forge has these files. So I'd suggest to add this channel to the environment.yml for Windows support.

Output on Windows:

foo@bar:~$ conda env create -f environment.yml

Collecting package metadata (repodata.json): done
Solving environment: failed

ResolvePackageNotFound:
  - cudatoolkit=11.1

@tljstewart
Copy link

tljstewart commented Feb 8, 2022

@SetZero but its interesting, because there readme says that a custom cuda kernel is used from Nvidia... does this change work for you?

Why is CUDA toolkit installation necessary?
The PyTorch package contains the required CUDA toolkit libraries needed to run PyTorch, so why is a separate CUDA toolkit installation required? Our models use custom CUDA kernels to implement operations such as efficient resampling of 2D images. PyTorch code invokes the CUDA compiler at run-time to compile these kernels on first-use. The tools and libraries required for this compilation are not bundled in PyTorch and thus a host CUDA toolkit installation is required.

https://github.com/NVlabs/stylegan3/blob/main/docs/troubleshooting.md#why-is-cuda-toolkit-installation-necessary

PDillis added a commit to PDillis/stylegan3-fun that referenced this pull request Mar 28, 2022
… for correctly building conda env in Windows; update README; add models in gen_utils.py; general code linting
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants