`nix develop` > Conda, venv, toolbox/distrobox

```

{

description = “Instant ROCM 5.7 environment”;

inputs.nixpkgs.url = "github:nixos/nixpkgs/nixos-23.11";

outputs = { self, nixpkgs }:

let

pkgs = import nixpkgs {

system = "x86_64-linux";

};

pythonPackages = pkgs.python310Packages;

pyPkgs = with nixpkgs; [

(pkgs.python310.withPackages(ps: with ps; [

pip

torchWithRocm

]))

];

in

{

devShells.x86_64-linux = {

default = pkgs.mkShell {

buildInputs = [

pyPkgs

pkgs.radeontop

pkgs.nvtop-amd

pkgs.rocmPackages.clr

pkgs.rocmPackages.hipcc

pkgs.rocmPackages.miopen

pkgs.rocmPackages.rocblas

pkgs.rocmPackages.hipblas

pkgs.rocmPackages.rocminfo

pkgs.rocmPackages.rocm-smi

pkgs.rocmPackages.llvm.llvm

pkgs.rocmPackages.llvm.clang

pkgs.rocmPackages.llvm.bintools

pkgs.rocmPackages.llvm.clang-tools-extra

];

};

};

};

}

```

Reply to this note

Please Login to reply.

Discussion

No replies yet.